D - How Many Answers Are Wrong——种类并查集

Think:
1知识点:种类并查集+后缀和数组
2思考:当学习新的知识点的时候不理解的地方应该动手实现在纸上模拟实现进而提高理解

vjudge题目链接

可参考博客1(种类并查集)
可参考博客2
可参考博客3

以下为Accepted代码

#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 2e5 + 4;

int n, m, f[N], sum[N];/*sum数组为后缀和数组:sum[x] = [ x, f[x] ]*/

void Init();
int get_f(int x);

int main(){
    int ans, i, l, r, w;
    while(~scanf("%d %d", &n, &m)){
        ans = 0;
        Init();
        for(i = 1; i <= m; i++){
            scanf("%d %d %d", &l, &r, &w);
            l -= 1;

            int t1 = get_f(l);
            int t2 = get_f(r);

            if(t1 == t2){
                if(sum[l]-sum[r] != w)
                    ans++;
            }
            else if(t1 < t2){
                f[t1] = t2;
                sum[t1] = sum[r] + w - sum[l];/*sum[t1] = [ t1, f[t1] ] = [t1, t2] = sum[r] + w - sum[l]*/
            }
            else if(t1 > t2){
                f[t2] = t1;
                sum[t2] = sum[l] - w - sum[r];/*sum[t2] = [ t2, f[t2] ] = [t2, t1] = sum[l] - w - sum[r]*/
            }
        }
        printf("%d\n", ans);
    }
    return 0;
}
void Init(){
    for(int i = 0; i <= n; i++){
        f[i] = i;
        sum[i] = 0;
    }
}
int get_f(int x){
    if(f[x] == x)
        return f[x];
    else {
        int t = f[x];
        f[x] = get_f(f[x]);
        sum[x] += sum[t];
        return f[x];
    }
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值