【带权并查集】HDU 3047 Zjnu Stadium

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

【题意】

http://blog.****.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+;
 const int inf=0x3f3f3f3f;
 int fa[maxn];
 int rk[maxn];
 void init()
 {
     for(int i=;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,,sizeof(rk));
         int A,B,x;
         int cnt=;
         for(int i=;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\n",cnt);
     }
     return ;
 }

带权并查集