Hihocoder-分数调查

题目名称：分数调查

题目链接：分数调查

描述

小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<bits/stdc++.h>
using namespace std;
#define N 100010

int f[N],score[N];
int n,m,q;
int x,y,s;

void init(){
	for(int i=1;i<=n;i++) {
		score[i]=0;
		f[i]=i;
	}
}

int find(int x){
	if(x==f[x]) return x;
	int t=f[x];
	f[x]=find(f[x]);
	score[x]+=score[t];
	return f[x];
}

void Union(){
	int fx=find(x);
	int fy=find(y);
	if(fx!=fy){
		f[fx]=fy;
		score[fx]=score[y]+s-score[x];
	}
}

int main()
{
	cin>>n>>m>>q;
	init();
	for(int i=0;i<m;i++){
		cin>>x>>y>>s;
		Union();
	}
	for(int i=0;i<q;i++){
		cin>>x>>y;
		if(find(x)!=find(y)) cout<<"-1"<<endl;
		else cout<<score[x]-score[y]<<endl;
	}
	return 0;
}