带权并查集

题目: link.

题意:

输入n和m 表示有n个数和m行输入,
接下来m行,每行输入x,y,w表示第x个数到第y个数的和是w,如果与之前的相矛盾,答案加+1,不矛盾则作为之后的条件。
比如1 10 10,1 5 4,6 10 5,前两条正确,第三条与则与之前的矛盾,因为1到10的和是10,1到5的和是4,6到10的和应该是6,而不是5,所以有一条是错误的,输出1。

思路:

很难想到这是一个并查集的题。
开一个sum数组存第i个点到根结点的和,初始化为0(自己是自己的根结点),每次find的时候更新到根结点的和,可以理解为前缀和,如果两个点不在一个并查集中,那么判断这两个点sum差值等不等于输入的和,不等于的话ans+1.
具体看代码

代码
#include <iostream>
#include <cstdio>
#include <map>
#include <set>
#include <cmath>
#include <queue>
#include <stack>
#include <unordered_set>
#include <vector>
#include <cstring>
#include <algorithm>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define pb push_back
#define T int T;scanf("%d",&T);while(T--)
const ll mod=1e9+7;
int f[200005];
int sum[200005];
int find(int x){   //查找的时候会更新sum的值
	if(x==f[x]) return x;
	int ff = find(f[x]);  
	sum[x] += sum[f[x]];  //一层层加回来,相当于前缀和
	f[x] = ff;
	return f[x];
}
int main(){
	int n,m;
	// scanf("%d %d", &n, &m);
	while(~scanf("%d %d", &n, &m)){
		for(int i = 0; i <= n+1; i ++){
			sum[i] = 0;
			f[i] = i;
		}
		int ans = 0;
		while(m--){
			int x,y;
			int w;
			scanf("%d %d %d", &x, &y, &w);
			y++;  //第x个点到第y个点的和 = sum[y+1]-sum[x],可以理解为前缀和
			int fx = find(x);	//查根结点的时候就更新了路上每个点sum值
			int fy = find(y);
			if(fx!=fy){
				f[fy] = fx;  //统一把小的节点作为根结点
				sum[fy] = sum[x] - sum[y] + w;  //把根结点连到另一个
			}else{
				if(w!=sum[y]-sum[x]) ans ++;
			}
		}
		printf("%d\n", ans);
	}

    return 0;
}
/*


*/

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值