How Many Answers Are Wrong(带权并查集)
题目:
多组输入数据。n,m。你不知道[1,n]内任意区间内值的和。 m次询问,a b 是端点,都在n的范围以内 ; v表示 [a,b]的区间内值的和。对每次询问,判断v是否与前面询问冲突,不冲突就当做是真的。
思路:
看上面的博客~
#include<iostream> #include<cstdio> #include <cctype> #include<algorithm> #include<cstring> #include<cmath> #include<string> #include<cmath> #include<set> #include<vector> #include<stack> #include<queue> #include<map> using namespace std; #define ll long long #define mem(a,x) memset(a,x,sizeof(a)) #define se second #define fi first const ll mod=998244353; const int INF= 0x3f3f3f3f; const int N=2e5+5; int n,m, a,b,v,fa,fb;; int f[N]; int sum[N]; int getf(int x) { if(x!=f[x]) { int t=f[x]; f[x]=getf( f[x]); //画图,压缩路径,加起来 sum[x]+=sum[t]; } return f[x]; } int main() { while(cin>>n>>m) { int cnt=0; for(int i=0;i<=n;i++){ sum[i]=0; f[i]=i; } while(m--) { scanf("%d%d%d",&a,&b,&v); a--; // !!! fa=getf(a); fb=getf(b); if(fa==fb) { if(sum[a]-sum[b]!=v) cnt++; } else { f[fa]=fb; sum[fa]=sum[b]+v-sum[a]; //画图,这里只用更新sum[fa]或者sum[fb]就行 ,其他的在路径压缩会自动更新 } } cout<<cnt<<endl; } }