并查集

How Many Answers Are Wrong

题目大意:
有一个数列长度为n,给出m个描述,每个表述给出一个区间和区间上的和,求这m个描述中有多少个和前面的是冲突的

解题思路:
我们不知道数列的具体内容,只知道若干区间和是多少,然而如果区间和是相邻的,则可以组合成一个更大的区间,也就是通过已有区间推理出更多区间和。
使用带权并查集,每一个节点表示一个区间的起点,它的父亲节点表示这个左闭右开区间的右端点,边上的权值就是这个区间上的和,对于给出的每一个描述先判断是否联通,如果联通就判断权值是否吻合,如果不连通就通过并查集合并操作判断是否联通。
在这里插入图片描述

#include <bits/stdc++.h>
using namespace std;

const int maxn = 200001;

int f[maxn];
int val[maxn];

int find(int t) {
    if (f[t] == -1)
        return t;
    int ans = find(f[t]);
    val[t] += val[f[t]]; //权值累加
    return f[t] = ans;
}

int main() {
    int n, m;
    while (scanf("%d%d", &n, &m) != EOF) {
        memset(f, -1, sizeof(f));
        memset(val, 0, sizeof(val));
        int ans = 0;
        while (m--) {
            int a, b, c;
            scanf("%d%d%d", &a, &b, &c);
            a = a - 1;
            int t1 = find(a);
            int t2 = find(b);
            if (t1 != t2) {
                f[t2] = t1;
                val[t2] = val[a] - val[b] + c;
            }
            else {
                if (val[b] - val[a] != c)
                    ans++;
            }
        }
        printf("%d\n", ans);
    }
    return 0;
}
#include <bits/stdc++.h>
using namespace std;

int fa[200010], value[200010];
int n, m; 

int findFa(int x) {
	if(fa[x] == x) return x;
	int new_fa = findFa(fa[x]);
	value[x] += value[fa[x]]; //权值需要累加,要加上父亲节点的权值
	fa[x] = new_fa; //路径压缩 
	return fa[x];
} 

int main() {
	while(cin >> n >> m) {
		int ans = 0;
		for(int i = 0; i < 200010; i ++) { //并查集初始化 
			fa[i] = i, value[i] = 0;
		}
		for(int i = 1, a, b, s; i <= m; i ++) {
			cin >> a >> b >> s;
			b++; //变成开区间
			if(findFa(a) == findFa(b)) { //如果两者在一个集合中,两点已经连通,正确答案可以推出来
				int t = value[b] + s;
				if(value[a] != t) //判断给的数据是否吻合
					ans ++;
				//ans += value[a] != value[b] + s;
			}
			else { //jin 
				value[fa[a]] = value[b] + s - value[a];
				fa[fa[a]] = fa[b];
				//边权可以是一个负数,代表从后往前的一个反向区间, 避免正负分类 
			} 
		}
		cout << ans << endl;
	}
	return 0;
}

“进阶算法”专栏(目录)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值