HDU 3047 带权并查集

点击打开链接

题意：给了n个人和m个关系，关系为a，b，x，意思是b的位置大于a的位置x，问条件依次下去的矛盾的条件的个数

思路：第一个带权的并查集，倒不是很难，它的算法思想非常给力，若两个不再同一集合的两个人，将b的根节点的父亲设为a的根节点，也就是合并，而所有的点到根的距离全部是用找根节点的路上完成更新的，而现在还需要的是b的根节点的位置，它的位置应该为r[bb]=r[a]+x-r[b]，r为到根节点的距离，a与b的位置相差x，而b与根节点的距离为r[b]，算一下这个式子就可以算出来了

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int maxn=50010;
int f[maxn],r[maxn];
void init(){
    for(int i=0;i<maxn;i++){
        f[i]=i;r[i]=0;
    }
}
int find1(int x){
    if(x==f[x]) return x;
    int t=f[x];
    f[x]=find1(f[x]);
    r[x]+=r[t];
    return f[x];
}
int unite(int a,int b,int x){
    int aa=find1(a);
    int bb=find1(b);
    if(aa==bb){
        if((r[a]+x)!=r[b]) return 1;
        else return 0;
    }else{
        f[bb]=aa;
        r[bb]=r[a]+x-r[b];
        return 0;
    }
}
int main(){
    int n,m,a,b,c;
    while(scanf("%d%d",&n,&m)!=-1){
        init();
        int ans=0;
        for(int i=0;i<m;i++){
            scanf("%d%d%d",&a,&b,&c);
            if(unite(a,b,c)) ans++;
        }
        printf("%d\n",ans);
    }
    return 0;
}