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

带权并查集

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

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

记 wi w i 表示根节点到 i i 这段时间的盈利，当x x 和 y y 不属于同一个根节点时，更新wrt[y]=wx+z−wy w r t [ y ] = w x + z − w y 并合并。当属于同一个根节点时判断 wy−wx w y − w x 是否 =z = z 。

代码：

#include
#include
#include
#include
#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;
}