Wormholes

第一次写spfa，写的有点乱，凑合着看吧。

 1 #include<cstdio>

 2 #include<cstring>

 3 #include<cstdlib>

 4 #include<algorithm>

 5 #include<iostream>

 6 #include<queue>

 7 #define MAXN 52000

 8 using namespace std;

 9 const int INF=1<<28;

10 bool vis[MAXN];

11 int head[MAXN];

12 int next[MAXN];

13 int dis[MAXN];

14 struct node

15 {

16     int u,v,w;

17 }p[MAXN];

18 int e,n,m;

19 int cnt[MAXN];

20 void addnode(int u,int v,int w)

21 {

22     p[e].u=u;

23     p[e].v=v;

24     p[e].w=w;

25     next[e]=head[u];

26     head[u]=e++;

27 }

28 bool relax(int u,int v,int w)

29 {

30     if(dis[v]>dis[u]+w)

31     {

32         dis[v]=dis[u]+w;

33         return true;

34     }

35     return false;

36 }

37 bool spfa(int rc)

38 {

39     memset(vis,false,sizeof(vis));

40     memset(cnt,0,sizeof(cnt));

41     for(int i=1;i<=n;i++)

42         dis[i]=INF;

43     dis[rc]=0;

44     vis[rc]=true;

45     queue<int>q;

46     q.push(rc);

47     while(!q.empty()){

48         int pre=q.front();

49         q.pop();

50         vis[pre]=false;

51         for(int i=head[pre];i+1;i=next[i])

52         {

53             if(relax(pre,p[i].v,p[i].w)&&!vis[p[i].v]){

54                 if((++cnt[p[i].v])>n) return false;

55                 q.push(p[i].v);

56                 vis[p[i].v]=true;

57             }

58         }

59     }

60     return true;

61 }

62 int main()

63 {

64     int t,w,ww,u,v;

65     cin>>t;

66     while(t--){

67         memset(head,-1,sizeof(head));

68         memset(next,-1,sizeof(next));

69         e=0;

70         cin>>n>>m>>ww;

71         for(int i=1;i<=m;i++)

72         {

73             cin>>u>>v>>w;

74             addnode(u,v,w);

75             addnode(v,u,w);

76         }

77         for(int i=1;i<=ww;i++)

78         {

79             cin>>u>>v>>w;

80             addnode(u,v,(-1*w));

81         }

82         if(!spfa(1))

83             printf("YES\n");

84         else

85             printf("NO\n");

86     }

87     return 0;

88 }

View Code