POJ3259(链式前向星+spfa)

#include<iostream>
#include<cstdio>
using namespace std;
int d[1001],p[1001],times[1001],q[1001*1001],to[10001],next[10001],begin[10001],w[10001];
int m,n,f,l,e;
void add(int x,int y,int z){
     to[++e]=y;
     next[e]=begin[x];
     begin[x]=e;
     w[e]=z;          
}

int main(){
    int i,j,k,t;
    int u;
    cin>>t;
    while(t--){    
	    int flag=0;            
        cin>>n>>m>>u;
        e=0;
        for(i=1;i<=n;i++)begin[i]=0;     
        for(i=1;i<=m;i++){
        	int x,y,z;
            cin>>x>>y>>z;                  
            add(x,y,z);
            add(y,x,z);
        }
        for(i=1;i<=u;i++){
        	int x,y,z;
            cin>>x>>y>>z;                
            add(x,y,-z);
        }    
        for(i=1;i<=n;i++){
            d[i]=100000001;
            p[i]=0;
            times[i]=0;
        }    
        f=0;l=1;
        d[1]=0;
		q[1]=1;
		p[1]=1;
		times[1]=1;
        while(f<l){
            f++;
            k=q[f];p[k]=0;            
            for(i=begin[k]; i ; i=next[i])
                if(d[to[i]]>d[k]+w[i]){
                    d[to[i]]=d[k]+w[i];
                    if(!p[to[i]]){
                        q[++l]=to[i];
                        p[to[i]]=1;
                        times[to[i]]++;                        
                        if(times[to[i]]>=n){
                            flag=1;                            
                            break;                    
                        }              
                    }                       
                }            
            if(flag==1)break;           
        }
        if(flag==1) cout<<"YES"<<endl;
        else cout<<"NO"<<endl;
    }    
    return 0;    
}

你可能感兴趣的:(C语言,poj,图论,SPFA,链式前向星)