HDU ~ 3038 ~ How Many Answers Are Wrong （带权并查集）

题意：有N个数字，M组关系。每组关系三个数字a,b,s表示a~b的和为s。问与前面产生矛盾的话有几组？
思路：带权并查集。多开一个权值数组，存储到自己和父节点的区间和。
图一：路径压缩，b~root的和 = b~a的和 + a ~ root的和。
图二：合并操作，现在我们知道a~root1和b~root2的区间和，又告诉了我们a~b的区间和，把root2并到root1上的话，

root1~root2的区间和为 （a~b）+ (b~root2) - （a~root1）。 
图三：求a~b的区间和，a~b的和 = （a~root） - （b ~ root）的和。



#include<bits/stdc++.h>
using namespace std;
const int MAXN = 200005;
int n, m, ans, f[MAXN], w[MAXN];//并查集数组和关系数组
//w[i]表示跟根节点的差
void init()
{
ans = 0;
for (int i = 0; i <= n; i++)
{
f[i] = i; w[i] = 0;
}
}
int Find(int x)
{
if (f[x] == x) return x;
int t = f[x];
f[x] = Find(f[x]);
w[x] = w[x] + w[t];
return f[x];
}
void Union(int a, int b, int s)
{
int root1 = Find(a), root2 = Find(b);
if (root1 != root2)
{
f[root1] = root2;
w[root1] = w[b] + s - w[a];
}
else
{
if (abs(w[a] - w[b]) != s) ans++;
}
}
int main()
{
while(~scanf("%d%d", &n, &m))
{
init();//初始化
while (m--)
{
int a, b, s;
scanf("%d%d%d", &a, &b, &s);
Union(a - 1, b, s);//a~b的区间和为w[b] - w[a - 1]
}
printf("%d\n", ans);
}
return 0;
}
/*
10 5
1 10 100
7 10 28
1 3 32
4 6 41
6 6 1
*/


4000