【bzoj 1202】[HNOI2005] 狡猾的商人（图论--带权并查集+前缀和）

 1 #include<cstdio>
 2 #include<cstdlib>
 3 #include<cstring>
 4 #include<iostream>
 5 using namespace std;
 6 
 7 const int N=105,M=1005;
 8 int n,m;
 9 int fa[N],f[N];
10 bool flag;
11 
12 int ffind(int x)
13 {
14     if (fa[x]!=x)
15     {
16       int fx=fa[x];
17       fa[x]=ffind(fx);
18       f[x]+=f[fx];
19     }
20     return fa[x];
21 }
22 void ins(int x,int y,int d)
23 {
24     int fx=ffind(x),fy=ffind(y);
25     if (fx!=fy)
26     {
27       fa[fx]=fy;//at ease
28       f[fx]=f[y]+d-f[x];
29     }
30     else if (f[x]-f[y]!=d) flag=false;
31 }
32 int main()
33 {
34     int T;
35     scanf("%d",&T);
36     while (T--)
37     {
38       int x,y,d;
39       scanf("%d%d",&n,&m);
40       flag=true;
41       for (int i=0;i<=n;i++) fa[i]=i,f[i]=0;
42       for (int i=1;i<=m;i++)
43       {
44         scanf("%d%d%d",&x,&y,&d);//s[y]-s[x-1]=d
45         if (flag) ins(x-1,y,d);
46       }
47       if (flag) printf("true\n");
48       else printf("false\n");
49     }
50     return 0;
51 }