【bzoj 1202】狡猾的商人（带权并查集）

传送门biu~
设$s_i$表示前$i$个月份的收入和。每条信息$(l,r,v)$相当于是在说$s_r-s_{l-1}=v$，可以用带权并查集来将$l-1$月份和$r$月份合并。设$root$月份为$i$月份所在的集合中的根，那么维护一个$dis_i$来表示$i$点到根的距离，即$dis_i=s_i-s_{root}$。
①当把两个不在同一个集合里的元素$x$和$y$用$v$合并时，设$x$的根为$xn$，$y$的根为$yn$，则$dis_x=s_x-s_{xn},dis_y=s_y-s_{yn},v=s_y-s_x$。如果用$xn$作为新集合的代表元素，那么$dis_{yn}=dis_x+v-dis_y$。
②假设$x$和$y$已经在同一个集合中，距离为$v$，只需要验证$dis_y$是否等于$dis_x+v$，如果不相等则不成立。

#include<bits/stdc++.h>
using namespace std;
struct data{int l,r,num;}a[1005];
int father[105],dis[105];
int search(int x){
    if(father[x]==x)    return x;
    int fa=search(father[x]);
    dis[x]+=dis[father[x]];
    return father[x]=fa;
}
int main(){
    int T;
    scanf("%d",&T);
    while(T--){
        int n,m;bool flag=false;
        scanf("%d%d",&n,&m);
        for(int i=0;i<=n;++i)   father[i]=i,dis[i]=0;
        for(int i=1;i<=m;++i)   scanf("%d%d%d",&a[i].l,&a[i].r,&a[i].num);
        for(int i=1;i<=m;++i){
            int x=a[i].l-1,y=a[i].r,z=a[i].num;
            int xn=search(x),yn=search(y);
            if(xn^yn)   dis[yn]=dis[x]+z-dis[y],father[yn]=xn;
            else if(dis[y]!=dis[x]+z){flag=true;break;}
        }
        printf(flag?"false\n":"true\n");
    }
    return 0;
}