POJ 3259 Wormholes（SPFA）

 

【题目链接】 http://poj.org/problem?id=3259

 

【题目大意】

　　给出一张n个点m条边的无向图，有w个虫洞，可以在穿越的过程中减少时间，问有没有可能回到过去

 

【题解】

　　回到过去即存在负权回路，SPFA求解即可
　　需要注意的是不是连通图，因此在做spfa的时候要将所有的点先入队一次

 

【代码】

#include <cstdio>
#include <cstring>
using namespace std;
const int N=200010,inf=~0U>>2,M=200000;
int x,S,T,time[N],q[N],size,h,t,n,m,W,ed,dis[N],in[N],nxt[N],w[N],v[N],g[N],u,e,cost;
void add(int x,int y,int z){v[++ed]=y;w[ed]=z;nxt[ed]=g[x];g[x]=ed;}
int spfa(int S){
    for(int i=1;i<=n;i++)dis[i]=inf,in[i]=0,time[i]=0;
    time[S]=1,dis[S]=0,in[S]=1;
    int i,x,size; q[h=t=size=1]=S;
    for(int i=1;i<=n;i++){time[i]++,t=(t+1)%M,size++,in[q[t]=i]=1;}
    while(size){
        for(i=g[x=q[h]],h=(h+1)%M,size--;i;i=nxt[i])if(dis[x]+w[i]<dis[v[i]]){
            dis[v[i]]=dis[x]+w[i];
            if(!in[v[i]]){
                time[v[i]]++,t=(t+1)%M,size++,in[q[t]=v[i]]=1;
                if(time[v[i]]>n)return -1;
            }
        }in[x]=0;
    }return dis[T];
}int cas;
int main(){
    scanf("%d",&cas);
    while(cas--){
        scanf("%d%d%d",&n,&m,&W);
        memset(v,0,sizeof(v)); memset(nxt,0,sizeof(nxt));
        memset(w,0,sizeof(w)); memset(g,0,sizeof(g)); ed=0;
        for(int i=1;i<=m;i++){
            scanf("%d%d%d",&u,&e,&cost);
            add(u,e,cost);
            add(e,u,cost);
        }
        for(int i=1;i<=W;i++){
            scanf("%d%d%d",&u,&e,&cost);
            add(u,e,-cost);
        }
        if(spfa(1)==-1)puts("YES"); 
        else puts("NO");
    }return 0;
}