首页 > 代码库 > 单源最短路 判负环

单源最短路 判负环

Wormholes http://poj.org/problem?id=3259

spfa 2e

 1 #include<cstdio> 2 #include<cstring> 3 #include<algorithm> 4 #include<queue> 5 #define mt(a,b) memset(a,b,sizeof(a)) 6 using namespace std; 7 const int inf=0x3f3f3f3f; 8 const int M=1024; 9 class Spfa { ///单源最短路o(2*ME)10     typedef int typec;///边权的类型11     static const int ME=1000010;///边的个数12     static const int MV=1000010;///点的个数13     struct E {14         int v,next;15         typec w;16     } e[ME];17     int n,le,head[MV],inque[MV];18     typec dist[MV];19     bool used[MV];20     queue<int> q;21 public:22     void init(int tn) { ///传入点的个数23         n=tn;24         le=0;25         mt(head,-1);26     }27     void add(int u,int v,typec w) {28         e[le].v=v;29         e[le].w=w;30         e[le].next=head[u];31         head[u]=le++;32     }33     bool solve(int s) { ///传入起点,下标0开始,存在负环返回false34         for(int i=0; i<n; i++) {35             dist[i]=inf;36             used[i]=true;37             inque[i]=0;38         }39         used[s]=false;40         dist[s]=0;41         inque[s]++;42         while(!q.empty()) q.pop();43         q.push(s);44         while(!q.empty()) {45             int u=q.front();46             q.pop();47             used[u]=true;48             for(int i=head[u]; ~i; i=e[i].next) {49                 int v=e[i].v;50                 if(dist[v]>dist[u]+e[i].w) {51                     dist[v]=dist[u]+e[i].w;52                     if(used[v]) {53                         used[v]=false;54                         q.push(v);55                         inque[v]++;56                         if(inque[v]>n) return false;57                     }58                 }59             }60         }61         return true;62     }63     typec getdist(int id) {64         return dist[id];65     }66 } gx;67 68 int main() {69     int t,n,m,w,x,y,z;70     while(~scanf("%d",&t)) {71         while(t--){72             scanf("%d%d%d",&n,&m,&w);73             gx.init(n);74             while(m--) {75                 scanf("%d%d%d",&x,&y,&z);76                 x--;77                 y--;78                 gx.add(x,y,z);79                 gx.add(y,x,z);80             }81             while(w--){82                 scanf("%d%d%d",&x,&y,&z);83                 x--;84                 y--;85                 gx.add(x,y,-z);86             }87             if(!gx.solve(0)) puts("YES");88             else puts("NO");89         }90     }91     return 0;92 }
View Code

 

 

end

单源最短路 判负环