第k短路

poj 2449 模板题  A*+spfa

  1 #include<iostream>
  2 #include<cstdio>
  3 #include<cstring>
  4 #include<queue>
  5 #define mt(a,b) memset(a,b,sizeof(a))
  6 using namespace std;
  7 const int inf=0x3f3f3f3f;
  8 class AStar { ///A*+spfa求第k短路
  9     typedef int typec;///边权的类型
 10     static const int ME=1e5+10;///边的个数
 11     static const int MV=1e3+10;///点的个数
 12     struct G {
 13         struct E {
 14             int v,next;
 15             typec w;
 16         } e[ME];
 17         int le,head[MV];
 18         void init(int n) {
 19             le=0;
 20             for(int i=0; i<=n; i++) head[i]=-1;
 21         }
 22         void add(int u,int v,typec w) {
 23             e[le].v=v;
 24             e[le].w=w;
 25             e[le].next=head[u];
 26             head[u]=le++;
 27         }
 28     };
 29     class Spfa { ///单源最短路o(k*ME)k~=2
 30         G g;
 31         int n,inque[MV],i,u,v;
 32         typec dist[MV];
 33         bool used[MV];
 34         queue<int> q;
 35     public:
 36         void init(int tn) { ///传入点的个数
 37             n=tn;
 38             g.init(n);
 39         }
 40         void add(int u,int v,typec w) {
 41             g.add(u,v,w);
 42         }
 43         bool solve(int s) { ///传入起点,存在负环返回false
 44             for(i=0; i<=n; i++) {
 45                 dist[i]=inf;
 46                 used[i]=true;
 47                 inque[i]=0;
 48             }
 49             used[s]=false;
 50             dist[s]=0;
 51             inque[s]++;
 52             while(!q.empty()) q.pop();
 53             q.push(s);
 54             while(!q.empty()) {
 55                 u=q.front();
 56                 q.pop();
 57                 used[u]=true;
 58                 for(i=g.head[u]; ~i; i=g.e[i].next) {
 59                     v=g.e[i].v;
 60                     if(dist[v]>dist[u]+g.e[i].w) {
 61                         dist[v]=dist[u]+g.e[i].w;
 62                         if(used[v]) {
 63                             used[v]=false;
 64                             q.push(v);
 65                             inque[v]++;
 66                             if(inque[v]>n) return false;
 67                         }
 68                     }
 69                 }
 70             }
 71             return true;
 72         }
 73         typec getdist(int id) {
 74             return dist[id];
 75         }
 76     } spfa;
 77     struct Q {
 78         int p;
 79         typec g,h;
 80         friend bool operator <(const Q &a,const Q &b) {
 81             return a.g+a.h>b.g+b.h;
 82         }
 83     } now,pre;
 84     priority_queue<Q> q;
 85     int n,cnt[MV];
 86     G g;
 87     typec ans;
 88 public:
 89     void init(int tn) {
 90         n=tn;
 91         g.init(n);
 92         spfa.init(n);
 93     }
 94     void add(int u,int v,typec w) {
 95         g.add(u,v,w);
 96         spfa.add(v,u,w);
 97     }
 98     bool solve(int s,int t,int k) {
 99         if(s==t) k++;
100         spfa.solve(t);
101         while (!q.empty()) q.pop();
102         for(int i=0; i<=n; i++) cnt[i]=0;
103         now.p=s;
104         now.g=0;
105         now.h=0;
106         q.push(now);
107         while(!q.empty()) {
108             pre=q.top();
109             q.pop();
110             int u=pre.p;
111             cnt[u]++;
112             if(cnt[u]==k&&u==t) {
113                 ans=pre.h+pre.g;
114                 return true;
115             }
116             if(cnt[u]>k) continue;
117             for(int i=g.head[u]; ~i; i=g.e[i].next) {
118                 now.h=pre.h+g.e[i].w;
119                 int v=g.e[i].v;
120                 now.g=spfa.getdist(v);
121                 now.p=v;
122                 q.push(now);
123             }
124         }
125         return false;
126     }
127     typec getans() {
128         return ans;
129     }
130 } gg;
131 int main() {
132     int n,m,u,v,w,s,t,k;
133     while(~scanf("%d%d",&n,&m)) {
134         gg.init(n);
135         while(m--) {
136             scanf("%d%d%d",&u,&v,&w);
137             gg.add(u,v,w);
138         }
139         scanf("%d%d%d",&s,&t,&k);
140         if(!gg.solve(s,t,k)) {
141             puts("-1");
142         } else {
143             printf("%d\n",gg.getans());
144         }
145     }
146     return 0;
147 }