题目: link.
题意:
输入n和m 表示有n个数和m行输入,
接下来m行,每行输入x,y,w表示第x个数到第y个数的和是w,如果与之前的相矛盾,答案加+1,不矛盾则作为之后的条件。
比如1 10 10,1 5 4,6 10 5,前两条正确,第三条与则与之前的矛盾,因为1到10的和是10,1到5的和是4,6到10的和应该是6,而不是5,所以有一条是错误的,输出1。
思路:
很难想到这是一个并查集的题。
开一个sum数组存第i个点到根结点的和,初始化为0(自己是自己的根结点),每次find的时候更新到根结点的和,可以理解为前缀和,如果两个点不在一个并查集中,那么判断这两个点sum差值等不等于输入的和,不等于的话ans+1.
具体看代码
代码
#include <iostream>
#include <cstdio>
#include <map>
#include <set>
#include <cmath>
#include <queue>
#include <stack>
#include <unordered_set>
#include <vector>
#include <cstring>
#include <algorithm>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define pb push_back
#define T int T;scanf("%d",&T);while(T--)
const ll mod=1e9+7;
int f[200005];
int sum[200005];
int find(int x){ //查找的时候会更新sum的值
if(x==f[x]) return x;
int ff = find(f[x]);
sum[x] += sum[f[x]]; //一层层加回来,相当于前缀和
f[x] = ff;
return f[x];
}
int main(){
int n,m;
// scanf("%d %d", &n, &m);
while(~scanf("%d %d", &n, &m)){
for(int i = 0; i <= n+1; i ++){
sum[i] = 0;
f[i] = i;
}
int ans = 0;
while(m--){
int x,y;
int w;
scanf("%d %d %d", &x, &y, &w);
y++; //第x个点到第y个点的和 = sum[y+1]-sum[x],可以理解为前缀和
int fx = find(x); //查根结点的时候就更新了路上每个点sum值
int fy = find(y);
if(fx!=fy){
f[fy] = fx; //统一把小的节点作为根结点
sum[fy] = sum[x] - sum[y] + w; //把根结点连到另一个
}else{
if(w!=sum[y]-sum[x]) ans ++;
}
}
printf("%d\n", ans);
}
return 0;
}
/*
*/