带权并查集（模板）

带权并查集

算法思想：维护节点到前导节点的距离；
HDU3047

#include <bits/stdc++.h>
using namespace std;
const int N=54321;
int dis[N],pre[N];
inline int find(int x)   //find（）函数，寻找根节点+路径压缩（前导节点，维护路径）；
{
    if(x==pre[x]) return x;
    int fa=pre[x];
    pre[x]=find(pre[x]); 
    dis[x]+=dis[fa];
    return pre[x];
}
int main()
{
    int n,m;
    while(scanf("%d%d",&n,&m)!=EOF){
        int i,x,y,fx,fy,val,tot=0;
        for(i=1;i<=n;++i){     //初始化；
            dis[i]=0;
            pre[i]=i;
        }
        while(m--){
            scanf("%d%d%d",&x,&y,&val);
            fx=find(x);                //x根节点；
            fy=find(y);                //y根节点；
            if(fx!=fy){               //没在一个集合里面（需要完成根节点之间的连接和权值维护）；
                pre[fy]=fx;
                dis[fy]=dis[x]+val-dis[y];    //dis[x]数组维护的是x到根节点的距离，x到y的距离，x到fx的距离，y到fy的距离，推到fx到fy的距离；
            }
            else if(dis[y]-dis[x]!=val){
                ++tot;
            }
        }
        printf("%d\n",tot);
    }
    return 0;
}