HihoCoder-1515-分数调查

 http://hihocoder.com/problemset/problem/1515

描述
小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
思路：这里记录的是两个点之间的差值，与节点值没有关系，因此不需要l--;

#include<iostream>

using namespace std;

const int N = 200000 + 10;

int dis[N]; //dis[i] 为 i 到 find(i) 的距离 
int p[N];

int find(int x)
{
	if (x != p[x])
	{
		int y = p[x];
		p[x] = find(p[x]);
		dis[x] += dis[y];
	}
	return p[x];
}

int main()
{
	int n, m, q;
	while (cin >> n >> m >> q)
	{
		for (int i = 0; i <= n; i++)
		{
			dis[i] = 0;
			p[i] = i;
		}
		while (m--)
		{
			int l, r, w;
			cin >> l >> r >> w;
			int pl = find(l);
			int pr = find(r);
			if(pl != pr)
			{
				p[pl] = pr;
				dis[pl] = dis[r] + w - dis[l];
			}			
		}
		while(q--)
		{
			int l, r;
			cin >> l >> r;
			int pl = find(l);
			int pr = find(r);
			if(pl == pr) cout<<dis[l] - dis[r]<<endl;
			else puts("-1");
		}
	}
	return 0;
}