#1515 : 分数调查（带权并查集）

Posted 2021-02-05 thunder-110

 

题目链接

 

题意：

　　小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;
            }
        }
    } 
}