带权并查集

HDU3047 Zjnu Stadium

最近一周好好把并查集和最短路走的尽量远一点 再然后就得往RMQ 差分约束 网络流方向走了

扯回来扯回来 刚写的一道带权并查集问题 (因为其实我的并查集基础很薄

并且再之前已经好几次碰到类似的问题了 就是说 带权并查集再合并两棵树的时候 他们各个节点是如何处理的 一直以来我都想不到好方法 今天刚写完这题 我才算明白算法的强大与互通性 看完其他人的博客 我马上意识到 —— 是在线段树里的 延迟更新 也就是说我先做个标记 当我访问的时候 我需要的时候再更新

而这里也是一个道理 我只对被合并的树 的根节点进行更新 当要访问这棵树的叶子节点 我再对其逐个节点递归更新 一直到访问节点

这个思路的博客其实是有很多的 但我实在是忍不住发一个 毕竟对我来说又是一大冲击

AC Code

#include <cstdio>
#include <cstring>
#include <cmath>
#include <queue>
#include <cstdlib>
#include <algorithm>

using namespace std;
const int maxn=50005;
int root[maxn],value[maxn];
int n,m;

void init()
{
    for(int i=0;i<=n;i++)
        root[i]=i,value[i]=0;
}

int fin(int x)
{
    if(root[x]==x) return x;
    int kao=root[x];
    root[x]=fin(root[x]);
    value[x]+=value[kao];//厉害
    return root[x];
}

bool Union(int a,int b,int x)
{
    int ra=fin(a);
    int rb=fin(b);
    if(ra==rb)
    {
        if(value[a]+x!=value[b]) return false;
        return true;
    }
    root[rb]=ra;
    value[rb]=value[a]+x-value[b];
    return true;
}

int main()
{
    int a,b,x;
    while(~scanf("%d%d",&n,&m))
    {
        init();
        int cnt=0;
        while(m--)
        {
            scanf("%d %d %d",&a,&b,&x);
            if(!Union(a,b,x))
                cnt++;
        }
        printf("%d\n",cnt);
    }
    return 0;
}

发表评论

电子邮件地址不会被公开。 必填项已用*标注