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

题目链接：How Many Answers Are Wrong

题目大意：

区间长度为n，给出m个区间和，即[l,r]区间和为w，对于给出的m个区间和，找到与上面区间和冲突的个数。

解题思路：

[l,r]=w,可以转化为v[r]-v[l-1]=w.

对于m个区间和，我们需要用带权并查集维护[l-1,r]的相对关系。

分两种情况：

1.r与l-1不在同一个集合，这是需要合并。

让fx指向fy，那么?+v[x]=w+v[y],即?=v[y]+w-v[x].

2.r与l-1在同一个集合,直接判断相对关系v[y]-v[x]==w，不相等则矛盾。

样例的图解

AC代码：

#include <bits/stdc++.h>
typedef long long ll;
using namespace std;
const int N = 2e5 + 10;
int f[N];
int v[N];
int n, m, ans;
void init() {  ///初始化
	ans = 0;
	for (int i = 0; i <= n; ++i) {
		f[i] = i; v[i] = 0;
	}
}
int find(int x) {  ///查找
	if (x != f[x]) {
		int t = f[x];
		f[x] = find(f[x]);
		v[x] += v[t];
	}
	return f[x];
}
void unite(int x, int y, int z) {   ///合并
	int fx = find(x), fy = find(y);
	f[fx] = fy;
	v[fx] = v[y] + z - v[x];
}
int main(void)
{
	while (cin >> n >> m) {
		init();
		while (m--) {
			int x, y, z; cin >> x >> y >> z;
			x--;
			if (find(x) == find(y)) {
				if (v[x] - v[y] != z) {
					ans++;
					cout << f[x] << " " << f[y] << endl;
					cout << v[x] << " " << v[y] << endl;
				}
			}
			else unite(x, y, z);
		}
		cout << ans << endl;
	}
	return 0;
}