帶權並查集（模板）

帶權並查集

算法思想：維護節點到前導節點的距離；
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;
}