POJ 2449 Astar算法 （A*） + Dijkstra

  1 #include <cstdio>
  2 #include <cstring>
  3 #include <algorithm>
  4 #include <iostream>
  5 #include <queue>
  6 #include <vector>
  7 using namespace std;
  8 
  9 #define N 1005
 10 #define inf 999999999
 11 
 12 struct mem{
 13     int y, w;
 14 };
 15 
 16 struct node{
 17     int x, g, h;
 18     bool operator<(node a)const{
 19         return a.g+a.h<g+h;          //以g+h最小的优先 出队 
 20     }
 21 };
 22 
 23 int visited[N];
 24 int dis[N];
 25 vector<mem>fan[N], zh[N];
 26 int n, m;
 27 int st, en, k;
 28 int cnt[N];
 29 
 30 void dijkstra(){
 31     int i, j, u, v, tmp;
 32     mem p, q;
 33     memset(visited,0,sizeof(visited));
 34     for(i=0;i<=n;i++) dis[i]=inf;
 35     dis[en]=0;
 36     for(i=1;i<=n;i++){     //刚开始我把end先压进去，然后i从2开始，但是会wa，改成这样就行了 
 37         int tmp=inf;
 38         for(j=1;j<=n;j++){
 39             if(dis[j]<tmp&&!visited[j]){
 40                 v=j;tmp=dis[j];
 41             }
 42         }
 43         if(tmp==inf) break;
 44         visited[v]=1;
 45         for(j=0;j<fan[v].size();j++){
 46             p=fan[v][j];
 47             if(!visited[p.y]&&dis[p.y]>dis[v]+p.w){
 48                 dis[p.y]=dis[v]+p.w;
 49             }
 50         }
 51     }
 52 }
 53 
 54 int Astar(){          //A*算法模板 
 55     
 56     int i, j;
 57     node p, q;
 58     mem pp;
 59     priority_queue<node>Q;
 60     memset(cnt,0,sizeof(cnt));
 61     while(!Q.empty()) Q.pop();
 62     p.x=st;p.g=0;p.h=dis[st];
 63     Q.push(p);
 64     while(!Q.empty()){
 65         p=Q.top();
 66         Q.pop();
 67         cnt[p.x]++;
 68         if(cnt[p.x]>k) continue; 
 69         if(cnt[en]==k) return p.g+p.h;   //当en出队k次即找到第k短路 
 70         for(i=0;i<zh[p.x].size();i++){
 71             pp=zh[p.x][i];
 72             q.x=pp.y;
 73             q.g=p.g+pp.w;
 74             q.h=dis[q.x];
 75             Q.push(q);
 76         }
 77     }
 78     return -1;
 79 }
 80 
 81 main()
 82 {
 83     int i, j;
 84     while(scanf("%d %d",&n,&m)==2){
 85         for(i=0;i<N;i++){
 86             zh[i].clear();
 87             fan[i].clear();
 88         }
 89         int x, y, z;
 90         mem p;
 91         while(m--){
 92             scanf("%d %d %d",&x,&y,&z);
 93             p.y=y;p.w=z;
 94             zh[x].push_back(p);
 95             p.y=x;
 96             fan[y].push_back(p);
 97         }
 98         scanf("%d %d %d",&st,&en,&k);
 99         if(st==en) k++;
100         dijkstra();
101     //    for(i=1;i<=n;i++) printf("%d ",dis[i]);
102         int ans=Astar();
103         printf("%d\n",ans);
104     }
105 }