#1515 : 分数调查(带权并查集)
题意:
小Hi的学校总共有N名学生,编号1-N。学校刚刚进行了一场全校的古诗文水平测验。 学校没有公布测验的成绩,所以小Hi只能得到一些小道消息,例如X号同学的分数比Y号同学的分数高S分。 小Hi想知道利用这些消息,能不能判断出某两位同学之间的分数高低?
思路:
学生a比学生b的分数高,可视为 b 是 a的父亲,a的权值为 v , 同理,假如c比b分数高,则b是c的父亲。 如果此时d比c的分数高,但是d不在并查集内,就把d加入并查集,因为d分数比c高,则箭头应该d->c,但是c的父亲是b,则d->b,即d的父亲是b。
如果两位同学不在同一并查集中,就不能判断。否则能判断,输出 a的权值和b的权值之差。
#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 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() { int n,m, a,b,v,fa,fb,q; while(cin>>n>>m>>q) { for(int i=0;i<=n;i++) sum[i]=0,f[i]=i; while(m--) { scanf("%d%d%d",&a,&b,&v); fa=getf(a); fb=getf(b); if(fa!=fb) { f[fa]=fb; sum[fa]=sum[b]+v-sum[a]; } } while(q--) { scanf("%d%d",&a,&b); fa=getf(a); fb=getf(b); if(fa!=fb){ cout<<-1<<endl; } else{ cout<<sum[a]-sum[b]<<endl; } } } }