hdu 3047 带权并查集

题目链接

相关知识

http://blog.csdn.net/shuangde800/article/details/7983965

题意

---

有n个人坐在zjnu体育馆里面，然后给出m个他们之间的距离， A B X， 代表B的座位比A多X. 然后求出这m个关系之间有多少个错误，所谓错误就是当前这个关系与之前的有冲突。

---

解析

---

1. 用带权并查集做， 对于并查集中的每一棵数， 树根的距离为0，然后以树根作为参照，每个结点的权值代表与树根的距离。
2. 合并A，B时，假设A，B属于不同的树，那么就要合并这两棵树， 把A树合并到B树上，这时要给A树的跟结点$root_a$赋值，关键是给$root_a$附上一个什么值。 由于A点和B点的权值dis[A]和dis[B]都是相对跟结点的距离，所以分析A，B之间的相对距离，可以得到$dis[root_a] = dis[A]+x-dis[B]$。 注意到这时，对于原来的A的树，只跟新了root_a跟结点的权值， 那么其它结点的跟新在查找的那一步里面实行了。

---

代码

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <algorithm>
#include <vector>
#include <cmath>
#include <map>
using namespace std;
const int inf=0x3f3f3f3f;
const int maxn = 100000+100;
int fa[maxn];
int dis[maxn];
void init() {
    for (int i=0; i<maxn; i++) 
        fa[i] = i, dis[i] = 0;
}

int Find(int x) {
    if (x == fa[x])
        return x;
    int t = Find(fa[x]);
    dis[x] += dis[fa[x]];
    return fa[x] = t;
}

int Union(int u, int v, int w) {
    int x=Find(u);
    int y=Find(v);
    if (x == y && dis[v] != dis[u]+w)
        return 1;
    fa[y] = x;
    dis[y] = dis[u]-dis[v]+w;
    return 0;
}

int main() {
    int n, m;
    while (~scanf("%d%d", &n, &m)) {
        init();
        int ans = 0;
        while (m--) {
            int u, v, w;
            scanf("%d%d%d", &u, &v, &w);
            if (Union(u, v, w))
                ans++;
        }
        printf("%d\n", ans);
    }
    return 0;
}