HDU 3047 Zjnu Stadium

http://acm.hdu.edu.cn/showproblem.php?pid=3047

并查集 记录各点到根的距离

#include <cstdio>
#include <cstring>
using namespace std;

const int N=50010,md=300;
int set[N],sum[N];
int find(int x)
{
    if(set[x]==x) return x;
    int t=find(set[x]);
    sum[x]+=sum[set[x]];
    return set[x]=t;
}
int main()
{
    int n,m;
    while(~scanf("%d%d",&n,&m))
    {
        memset(sum,0,sizeof(sum));
        for(int i=1;i<=n;i++) set[i]=i;
        int cnt=0;
        while(m--)
        {
            int a,b,x;
            scanf("%d%d%d",&a,&b,&x);
            int fa=find(a), fb=find(b);
            if(fa==fb)
            {
                int t=(sum[b]-sum[a])%md;
                if(t<0) t+=md;
                if(t!=x) cnt++;
            }
            else
            {
                set[fb]=fa;
                sum[fb]=x+sum[a]-sum[b];
            }
        }
        printf("%d\n",cnt);
    }
    return 0;
}