How Many Answers Are Wrong,权值并差集。题目意思就是,给定区间和,判断是否正确,统计不正确的数量,题目没有说数的正负性,所以可能是正,可能是负。只有当不相同的时候,才是不正确的。权值并差集在getf的时候,需要计算出当前节点到根节点的权值。先记录当前节点父亲编号。当找到根节点后,开始回溯时,每个节点的权值=此节点的权值+此节点父亲节点的权值。在合并的时候,如果2个区间都在同一个集合,直接查询区间值与给出值判定,如果不相同说明不正确。区间不在一个集合,那么就合并,合并路径,合并权值。权值合并到前一个sum[x1]=-sum[l]+sum[r]+val;细节就是,为了使[1,2][3,4]能合并起来,左区间都-1.
AC代码:
#include <cstdio>
#include <cstring>
#include <string>
#include <iostream>
#include <algorithm>
#include <set>
#include <map>
#include <stack>
#include <queue>
#include <deque>
#include <vector>
#include <cmath>
#include <functional>
#define il inline
#define INF (1<<30)
#define rep(i,x,y) for(int i=(x);i<=(y);i++)
#define down(i,x,y) for(int i=(x);i>=(y);i--)
using namespace std;
int const N=200000+100;
int n,m,l,r,val;
int a[N],sum[N];
int getf(int v) {
if(v!=a[v]) {
int t=a[v];
a[v]=getf(a[v]);
sum[v]+=sum[t];
}
return a[v];
}
int main() {
while(~scanf("%d%d",&n,&m)) {
for(int i=0; i<=n; i++) {
sum[i]=0;
a[i]=i;
}
int ans=0;
while(m--) {
scanf("%d%d%d",&l,&r,&val);
l--;
int x1=getf(l);
int x2=getf(r);
if(x1!=x2) {
a[x1]=x2;
sum[x1]=-sum[l]+sum[r]+val;
} else {
if(sum[l]-sum[r]!=val) {
ans++;
}
}
}
printf("%d\n",ans);
}
return 0;
}