题意
给定一段区间的和为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--;
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;
}