HDU 3038 How Many Answers Are Wrong(带权并查集)

时间:2023-03-08 17:47:15

太坑人了啊,读入数据a,b,s的时候,我刚开始s用的%lld,给我WA。 实在找不到错误啊,后来不知怎么地突然有个想法,改成%I64d,竟然AC了

思路:我建立一个sum数组,设i的父亲为fa,sum[i]表示(fa,i]中的数的和(不包括fa,包括i),

  合并的时候,不是合并a,b,而是合并a-1,b。这样做的目是因为s是[a,b]的和,如果直接合并a,b,那么按照我数组的定义应该是(a,b]的和,这样不符合题意。

  接下来,每次读入a,b,只要根据他们父节点的不同情况分类讨论即可。

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std; const int maxn=;
int father[maxn];
long long sum[maxn]; //设i的父亲为fa,sum[i]表示(fa,i]中的数的和(不包括fa,包括i)
int n,m;
void init() {
for(int i=; i<=n; i++) {
father[i]=i;
sum[i]=;
}
} int find_root(int x) {
if(father[x]==x)
return x;
int fa=father[x];
father[x]=find_root(father[x]);
sum[x]=(sum[x]+sum[fa]);
return father[x];
}
//合并的时候,保证x<y
void Union(int x,int y) {
father[y]=x;
}
int main() {
int a,b,ans=;
long long s;
while(scanf("%d%d",&n,&m)!=EOF) {
init(); //怎么老忘记写这个啊
ans=;
for(int i=; i<=m; i++) {
scanf("%d%d%I64d",&a,&b,&s); //就是这里,本来用的是lld的格式,导致WA
int fa=find_root(a-);
int fb=find_root(b);
/*
fa=fb时,(fa,a-1]=sum[a-1],(fb,b]=sum[b],
——>[a,b]=sum[b]-sum[a-1],显然即应该等于s
若不等于,即使错误的
*/
if(fa==fb) {
if(sum[b]-sum[a-]!=s) {
ans++;
}
}
else {
if(fb>a-) {
/*
大小关系:fa<=a-1<fb<=b
(fa,a-1]=sum[a-1],[a,b]=s,(fb,b]=sum[b];
——>[a,fb]=s-sum[b],
(fa,fb]=(fa,a-1]+[a-1,fb]=sum[a-1]+s-sum[b];
*/
Union(fa,fb);
//更新fb的父节点fa到fb区间的和(不包括fa)
sum[fb]=sum[a-]+s-sum[b];
} else if(fb==a-) {
/*
大小关系:fa<=a-1=fb<=b
(fa,a-1]=sum[a-1],[a,b]=s,(fb,b]=sum[b];
——>sum[b]即为[a,b]的和。
若sum[b]不等于s,说明错误
如果相同,则合并
*/
if(s!=sum[b]) {
ans++;
} else {
Union(fa,fb);
sum[fb]=sum[a-];
}
}
//fb<a-1
else {
if(fb>fa) {
/*
大小关系:fa<fb<a-1<b
(fb,b]=sum[b],(a-1,b]=s ——>(fb,a-1]=sum[b]-s;
再由(fa,a-1]=sum[a-1]——>(fa,fb]=sum[a-1]-sum[b]+s;
*/
Union(fa,fb);
sum[fb]=sum[a-]-sum[b]+s;
}
else {
/*
大小关系:fb<fa<=a-1<=b
(fb,b]=sum[b],(a-1,b]=s,(fa,a-1]=sum[a-1];
——>(fa,b]=s+sum[a-1]
——>(fb,fa]=sum[b]-s-sum[a-1];
*/
Union(fb,fa);
sum[fa]=sum[b]-s-sum[a-];
} }
}
}
printf("%d\n",ans);
}
return ;
}