题解:
没看出来要用并查集做,很不错的一道题~
区间[Ai,Bi]可以理解为前缀和相减sum(Bi)-sum(Ai-1),用每个顶点记录前缀和。如一条数据Ai Bi Ci,如果Ai,Bi不在同一个集合,则合并Ai,Bi所在集合并更新关系rel。关系rel[a]记录的是顶点a到根节点x的区间和。
代码:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int Max_n=2e5+10;
int n,m;
int par[Max_n];
int rel[Max_n];
int Find(int x){
if(par[x]==x)return x;
else {
int p=x;
x=Find(par[x]);
rel[p]=rel[p]+rel[par[p]];
return par[p]=x;
}
}
int main()
{
while(~scanf("%d%d",&n,&m)){
for(int i=0;i<=n;i++){
par[i]=i;
rel[i]=0;
}
int a,b,c,ans=0;
for(int i=0;i<m;i++){
scanf("%d%d%d",&a,&b,&c);
int x=Find(a-1),y=Find(b);
if(x!=y){
par[y]=x;
rel[y]=rel[a-1]+c-rel[b];
}
else if(rel[a-1]+c!=rel[b])ans++;
}
printf("%d\n",ans);
}
return 0;
}