hdu 3047 Zjnu Stadium

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

带权并差集

 #include <cstdio>
 #include <cstring>
 #include <algorithm>
 #define maxn 60000
 using namespace std;

 int f[maxn],d[maxn];
 int ans=;
 int n,m;

 int find1(int x)
 {
     if(x==f[x]) return x;
     int f1=f[x];
     f[x]=find1(f[x]);
     d[x]+=d[f1];
     return f[x];
 }

 void  union1(int a,int b,int x)
 {
     int fa=find1(a);
     int fb=find1(b);
     if(fa!=fb)
     {
         f[fb]=fa;
         d[fb]=d[a]+x-d[b];
         return;
     }
     if(fa==fb)
     {
         if(d[b]-d[a]!=x) ans++;
     }
 }

 void inti()
 {
     for(int i=; i<=n; i++)
     {
         f[i]=i;
     }
     memset(d,,sizeof(d));
     ans=;
 }

 int main()
 {
     while(scanf("%d%d",&n,&m)!=EOF)
     {
         inti();
         int a1,b1,x1;
         for(int i=; i<=m; i++)
         {
             scanf("%d%d%d",&a1,&b1,&x1);
             union1(a1,b1,x1);
         }
         printf("%d\n",ans);
     }
     return ;
 }