BZOJ1202 [HNOI2005]狡猾的商人（洛谷P2294）

带权并查集

BZOJ题目传送门
洛谷题目传送门

之前做过几乎一样的题目（这里），然而忘光了

记$w_i$表示根节点到$i$这段时间的盈利，当$x$和$y$不属于同一个根节点时，更新$w_{rt[y]}=w_x+z-w_y$并合并。当属于同一个根节点时判断$w_y-w_x$是否$=z$。

代码：

#include<cctype>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 105
using namespace std;
inline char readc(){
    static char buf[100000],*l=buf,*r=buf;
    if (l==r) r=(l=buf)+fread(buf,1,100000,stdin);
    return l==r?EOF:*l++;
}
inline int _read(){
    int x=0,f=1; char ch=readc();
    while (!isdigit(ch)) { if (ch=='-') f=-1; ch=readc(); }
    while (isdigit(ch)) x=(x<<3)+(x<<1)+(ch^48),ch=readc();
    return x*f;
}
int fa[N],w[N];
int findfa(int x){
    if (fa[x]==x) return x; int f=findfa(fa[x]);
    w[x]+=w[fa[x]]; return fa[x]=f;
}
int main(){
    for (int t=_read();t;t--){
        int n=_read(),m=_read(); bool f=true;
        for (int i=0;i<=n;i++) fa[i]=i,w[i]=0;
        while (m--){
            int x=_read()-1,y=_read(),z=_read(),fx,fy;
            if ((fx=findfa(x))!=(fy=findfa(y)))
                w[fy]=z+w[x]-w[y],fa[fy]=fx;
            else if (w[y]-w[x]!=z) f=false;
        }
        puts(f==true?"true":"false");
    }
    return 0;
}