[Offer收割]编程练习赛15 B.分数调查[加权并查集]

#1515 : 分数调查

时间限制:10000ms

单点时限:1000ms

内存限制:256MB

描述

小Hi的学校总共有N名学生，编号1-N。学校刚刚进行了一场全校的古诗文水平测验。  

学校没有公布测验的成绩，所以小Hi只能得到一些小道消息，例如X号同学的分数比Y号同学的分数高S分。  

小Hi想知道利用这些消息，能不能判断出某两位同学之间的分数高低？

输入

第一行包含三个整数N, M和Q。N表示学生总数，M表示小Hi知道消息的总数，Q表示小Hi想询问的数量。  

以下M行每行三个整数，X, Y和S。表示X号同学的分数比Y号同学的分数高S分。  

以下Q行每行两个整数，X和Y。表示小Hi想知道X号同学的分数比Y号同学的分数高几分。  

对于50%的数据，1 <= N, M, Q <= 1000  

对于100%的数据，1 <= N, M, Q<= 100000 1 <= X, Y <= N -1000 <= S <= 1000

数据保证没有矛盾。

输出

对于每个询问，如果不能判断出X比Y高几分输出-1。否则输出X比Y高的分数。

样例输入

10 5 3  
1 2 10  
2 3 10  
4 5 -10  
5 6 -10  
2 5 10  
1 10  
1 5  
3 5

样例输出

-1  
20  
0

#include<cstdio>
using namespace std;
const int N=1e5+5;
int n,m,q,fa[N],dis[N];
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
int find(int x){
    int t;
    if(fa[x]!=x) t=find(fa[x]),dis[x]+=dis[fa[x]],fa[x]=t;
    return fa[x];
}
int main(){
    n=read();m=read();q=read();
    for(int i=1;i<=n;i++) fa[i]=i;
    for(int i=1,l,r,s,u,v;i<=m;i++){
        r=read();l=read();s=read();
        u=find(l);v=find(r);
        if(u!=v) dis[u]=dis[r]+s-dis[l],fa[u]=v;
    }
    for(int i=1,x,y;i<=q;i++){
        x=read();y=read();
        if(find(x)!=find(y)) puts("-1");
        else printf("%d\n",dis[y]-dis[x]);
    }
    return 0;
}