并查集 带权并查集

描述
小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

本题目是一个带权并查集问题,tcl,一开始想成了最短路,对于本题而言,带权并查集的权值是当前点和根节点的差值,更改权值的操作应该是在合并的时候,因为合并时根节点会相应的发生变化,首先是将一个根并到另一个根上,有一个根节点变成了非根节点,因此会有权值的变化,首先本来根节点的子节点已经记录了当前节点到根的差值,将根并到另一个根上只需要记录一下变化的根到不变的根的差值,然后在 find 函数里递归查找的时候,对于每一个点,加上前一个根节点的权值即可。

ps:应该先更改再合并,因为合并前的节点信息才是应该被加上的

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <vector>
#include <queue>
using namespace std;
const int maxn = 1e5 + 10;
int n, m, q;
int root[maxn];
int cha[maxn];
int find(int a){
	if(a == root[a]){
		return a;
	}
	int t = find(root[a]);//ps
	cha[a] += cha[root[a]];//ps
	root[a] = t;//ps
	return root[a];
}
void mer(int a, int b, int c){
	int fa = find(a);
	int fb = find(b);
	if(fa != fb){
		root[fa] = fb;
		cha[fa] = c + cha[b] - cha[a];
	}
}
int main()
{
	scanf("%d %d %d", &n, &m, &q);
	memset(cha, 0, sizeof cha);
	for (int i = 0; i < maxn; i++)root[i] = i;
	for (int i = 1; i <= m; i++){
		int a, b, c;
		scanf("%d %d %d", &a, &b, &c);
		mer(a, b, c);
	}
	while (q--){
		int a, b;
		scanf("%d %d", &a, &b);
		if(find(a) != find(b))puts("-1");
		else{
			printf("%d\n", cha[a] - cha[b]);
		}
	}
	return 0;
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值