HDU 3038 How Many Answers Are Wrong 带权并查集

最近一连做了好几道就着区间建图的图论题，发现建图的时候都很喜欢把区间看成两个前缀和之差，我小叮当长见识了。这道题也是这样。一开始的时候想让r数组表示当前点到其祖先这个区间（包含两端点），后来要路径压缩了，发现情况不大对头。。。前缀和大法好！果然应该记录当前点与祖先的前缀和之差！

#include<iostream>
#include<cstdio>
#include<string.h>
#include<algorithm>
using namespace std;

int n,m,fa[200010],r[200010];

int find(int x)
{
    int fx=fa[x];
    if (fa[x]!=x)
    {
        fx=find(fa[x]);
        r[x]+=r[fa[x]];
    }
    return fa[x]=fx;
}

bool u(int x,int y,int v)
{
    int fx=find(x);
    int fy=find(y);
    if (fx==fy)
    {
        if (r[y]-r[x]==v) return true;
        else return false;
    }
    fa[fy]=fx;
    r[fy]=r[x]+v-r[y];
    return true;
}

int main()
{
    while (~scanf("%d%d",&n,&m))
    {
        for (int i=0; i<=n; i++)
        {
            fa[i]=i;
            r[i]=0;
        }
        int ans=0;
        for (int i=1; i<=m; i++)
        {
            int a,b,s;
            scanf("%d%d%d",&a,&b,&s);
            a--;
            if (!u(a,b,s)) ans++;
        }
        cout<<ans<<endl;
    }
    return 0;
}