【带权并查集】HDU 3047 Zjnu Stadium

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

【题意】

http://blog.csdn.net/hj1107402232/article/details/9921311

【Accepted】

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<algorithm>
#include<queue>
#include<stack>
#include<map>
using namespace std;

int n,m;
const int maxn=5e4+2;
const int inf=0x3f3f3f3f;
int fa[maxn];
int rk[maxn];
void init()
{
    for(int i=1;i<=n;i++)
    {
        fa[i]=i;
    }
}

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

void mix(int x,int y,int d)
{
    int fx=find(x);
    int fy=find(y);
    if(fx!=fy)
    {
        rk[fy]=rk[x]-rk[y]+d;
        fa[fy]=fx;
    }
}

int query(int x,int y)
{
    int fx=find(x);
    int fy=find(y);
    if(fx==fy)
    {
        return rk[y]-rk[x];
    }
    else
    {
        return inf;
    }
}

int main()
{
    while(~scanf("%d%d",&n,&m))
    {
        init();
        memset(rk,0,sizeof(rk));
        int A,B,x;
        int cnt=0;
        for(int i=0;i<m;i++)
        {
            scanf("%d%d%d",&A,&B,&x);
            int d=query(A,B);
            if(d==inf) 
            {
                mix(A,B,x);
            }
            else if(d!=x)
            {
                cnt++;
            }
        }
        printf("%d
",cnt);
    }
    return 0;
}