HihoCoder 1515 分数调查

题目描述：

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

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

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

Input

第一行包含三个整数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

数据保证没有矛盾。

Output

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

Sample Input

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

Sample Output

-1  
20  
0

 

思路：带权并查集，tree[y]=x表示x比y大，用权来表示点之间的相对数量关系，value[y]=n，表示x比y大n

#include <stdlib.h>
#include <stdio.h>
#include <string.h>

int n,m,q;
int tree[100010];
int value[100010];

int find(int x){
    int tmp=x;
    while(tree[tmp]!=tmp){
        tmp=tree[tmp];
        value[x]+=value[tmp];
    }
    return tree[x]=tmp;
}

int main()
{
    int x,y,s;
    scanf("%d%d%d",&n,&m,&q);
    memset(value,0,sizeof(value));
    for(int i=0;i<=n;i++){
        tree[i]=i;
    }
    while(m--){
        scanf("%d%d%d",&x,&y,&s);
        int rx=find(x);
        int ry=find(y);
        if(rx!=ry){
            tree[ry]=rx;
            value[ry]=value[x]+s-value[y];
        }
    }
    while(q--){
        scanf("%d%d",&x,&y);
        int rx=find(x);
        int ry=find(y);
        if(rx!=ry) printf("-1\n");
        else printf("%d\n",value[y]-value[x]);
    }
    return 0;
}