HihoCoder - 1515 分数调查（带权并查集）

#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

题意：中文题，就不说了

思路：简单的带权并查集，在写带权并查集的时候，要注意val【i】的值到底是父亲比儿子多还是儿子比父亲多，不论如何都可以，只要至始至终都保证是统一的即可。

解释一下注释（1）处的代码：

x的父亲是a，y的父亲是b，现在使得a的父亲为b，即把这两个集合合并，需要修改val【a】的值

x=a+val[x] 

y=b+val[y]

两个式子相减得 x-y=a-b+val[x]-val[y]

即 s=val[a]+val[x]-val[y]

化简得到val[a]=s-val[x]+val[y]

#include<cstdio>
#include<cstring>
#include<cmath>
#include<iostream>
#include<algorithm>
#define maxn 100005
using namespace std;
typedef long long ll;

int fa[100005];
int val[100005];

int findfa(int x)
{
	if(fa[x] == x) return x;
	int tmp=fa[x];
	fa[x]=findfa(fa[x]);
	val[x]+=val[tmp];//儿子比父亲多val
	return fa[x]; 
}


int main(){
	int n,m,q;
	scanf("%d%d%d",&n,&m,&q);
	int i,j;
	for(i=1;i<=n;i++){
		fa[i]=i;
		val[i]=0;
	}
	for(i=1;i<=m;i++){
		int x,y,s;
		scanf("%d%d%d",&x,&y,&s);
		int a=findfa(x);
		int b=findfa(y);
		if(a!=b){
			fa[a]=b;
			val[a]=s-val[x]+val[y];//（1）
		}
	}
	while(q--){
		int x,y;
		scanf("%d%d",&x,&y);
		int a=findfa(x);
		int b=findfa(y);
		if(a!=b){
			printf("-1\n");
		}
		else{
			printf("%d\n",val[x]-val[y]);
		}
	}
	return 0;
}