HDU 3047 Zjnu Stadium 帶權並查集

衆所周知，我們做這種種類並查集，喜歡開一個數組，用向量的思想去記錄和計算當前點和它祖先的關係。這個題也一樣，按部就班把式子寫上就行了，建圖也很容易想。

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

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

int find(int x)
{
    int fx=fa[x];
    if (fa[x]!=x)
    {
        fx=find(fa[x]);
        r[x]=(r[x]+r[fa[x]])%300;
    }
    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]+300)%300==v) return true;
        else return false;
    }
    fa[fy]=fx;
    r[fy]=(r[x]+v-r[y]+300)%300;
    return true;
}

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