HDU - 3038 -- How Many Answers Are Wrong【带权并查集】

How Many Answers Are Wrong

题意

给定一段区间的和为x，问当前区间的和与前面的区间有没有冲突，如果有则是错误答案。最后输出错误答案数。
	eg: 
	[1, 10] = 5
	[1, 3] = 2
	[4, 10] = 4
	很显然第三个区间就是错误的，因为[1,3]+[4,10] != [1,10].

思路

怎样判断当前区间与前面的区间形成冲突？ 我们需要借助一个载体(根节点)，记录左端点到其根节点的和与右端点到根节点的和，
我们以集合的最左端为根节点，判断以下三种情况。 
	1.l的根节点和r的根节点相同， 判断sum[r] - sum[l] == w; w为sum[r - l]
	2.l的根节点大于r的根节点，l的根节点认其r的根节点为根结点。 
	更新sum[l的根节点] = sum[r] - sum[l] - w; 
	3.r的根节点大于l的根节点，r的根节点认其l的根节点为根结点。 
	更新sum[r的根节点] = sum[l] - sum[r] + w;
根据区间合并原理，我们将左端点-1处理成闭区间，以上面的例子为例，处理为(0,3] + (3, 10] = (0,10] == [1,10].  

AC代码

#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn = 200005;
int f[maxn], sum[maxn];
int find(int x){
	if (x == f[x])	return f[x];
	int k = find(f[x]);
	sum[x] += sum[f[x]];//进行和的更新
	return f[x] = k;
}
void init(int n){
	for (int i = 0; i <= n; ++i){
		f[i] = i;
		sum[i] = 0;
	}
}
void solve(){
	int n, m;
	while(~scanf("%d%d", &n, &m)){
		init(n);
		int l, r, w, ans = 0;
		while (m--){
			scanf("%d%d%d", &l, &r, &w);
			l--;//根据区间合并原理，左端点需-1. 
			int sl = find(l);
			int sr = find(r);
			if (sl == sr && sum[r] - sum[l] != w){
				ans++;
			} 
			//进行合并时，认最左端点为父亲。
			else if (sl > sr){
				f[sl] = sr;
				sum[sl] = sum[r] - sum[l] - w;
			}
			else if (sl < sr){
				f[sr] = sl;
				sum[sr] = sum[l] - sum[r] + w;
			}	
		}
		printf("%d\n", ans);
	}
}
int main(){
	solve();
	return 0;
}