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