POJ 3259 Wormholes 虫洞（负权最短路，负环）

 1 //#include <bits/stdc++.h>
 2 #include <iostream>
 3 #include <cstdio>
 4 #include <cstring>
 5 #include <vector>
 6 #include <deque>
 7 #define LL long long
 8 #define pii pair<int,int>
 9 #define INF 0x7f7f7f7f
10 using namespace std;
11 const int N=550;
12 int n, m, edge_cnt;
13 vector<int> vect[N];
14 struct node
15 {
16     int from, to, val;
17     node(){};
18     node(int f,int t,int v):from(f),to(t),val(v){};
19 }edge[600000];
20 
21 
22 void add_node(int from,int to,int val)
23 {
24     edge[edge_cnt]=node(from,to,val);
25     vect[from].push_back(edge_cnt++);
26 }
27 
28 int dis[N], inq[N], cnt[N];
29 int spfa(int s)//模板
30 {
31     memset(inq,0,sizeof(inq));
32     memset(cnt,0,sizeof(cnt));
33     memset(dis,0x7f,sizeof(dis));
34     deque<int> que(1,s);
35     inq[s]=1;
36     dis[s]=0;
37 
38     while(!que.empty())
39     {
40         int x=que.front();
41         que.pop_front();
42         inq[x]=0;
43         for(int i=0; i<vect[x].size(); i++)
44         {
45             node e=edge[vect[x][i]];
46             if(dis[e.to]>dis[x]+e.val)
47             {
48                 dis[e.to]=dis[x]+e.val;
49                 if(!inq[e.to])
50                 {
51                     if(++cnt[e.to]>n)   return false;
52                     inq[e.to]=1;
53                     que.push_back(e.to);
54                 }
55             }
56         }
57     }
58     return true;
59 }
60 
61 int main()
62 {
63     freopen("input.txt", "r", stdin);
64     int a, b, c, t, w;
65     cin>>t;
66     while(t--)
67     {
68         scanf("%d%d%d",&n,&m,&w);
69         edge_cnt=0;
70         memset(edge,0,sizeof(edge));
71         for(int i=0; i<=n; i++) vect[i].clear();
72 
73         for(int i=0; i<m; i++)  //无向边
74         {
75             scanf("%d%d%d",&a,&b,&c);
76             add_node(a,b,c);
77             add_node(b,a,c);
78         }
79         vector<int> ver;
80         for(int i=0; i<w; i++)  //有向
81         {
82             scanf("%d%d%d",&a,&b,&c);
83             add_node(a,b,-c);
84             ver.push_back(a);
85         }
86         int i;
87         for(i=0; i<ver.size(); i++)
88         {
89             int q=ver[i];
90             if(!spfa(q))
91             {
92                 puts("YES");
93                 break;
94             }
95         }
96         if(i==ver.size())   puts("NO");
97     }
98     return 0;
99 }