带权并查集

HDU 2818 && POJ1988 Cube Stacking

Posted on

今天打比赛前写的一道带权并查集

题意 搬砖头 把包含有 a的砖头堆里的 所有砖头 转移到 包含b的砖头堆上 问 K砖头下有几块砖头

思路 :并查集 额外开两个数组 一个记录这堆砖头堆的砖头高度 另一个记录k砖头下有几块砖头 每次合并的时候更新砖头高度 和被合并的砖头堆 最底下砖头 下面有几块砖头 用户find 时更新

AC Code

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

using namespace std;
const int maxn=30005;
int root[maxn],under[maxn],height[maxn];
void init(int n)
{
    for(int i=0;i<=n;i++)
        root[i]=i,under[i]=0,height[i]=1;
}

int fin(int x)
{
    if(root[x]==x) return x;
    int rot=root[x];
    root[x]=fin(root[x]);
    under[x]+=under[rot];//递归更新要查找的点
    return root[x];
}

void Union(int u,int v)
{
    int ru=fin(u);
    int rv=fin(v);
    if(ru==rv) return;
    root[ru]=rv;//ru堆的最底下变成了rv
    under[ru]=height[rv];//ru砖头下有rv块砖头
    height[rv]+=height[ru];//更新rv堆砖头高度
}

int main()
{
    char ch[5];
    int u,v,n;
    while(scanf("%d",&n)!=EOF)
    {
        init(n<30000?n:30000);
        for(int i=0;i<n;i++)
        {
            scanf("%s",ch);
            if(ch[0]=='M')
            {
                scanf("%d%d",&u,&v);
                Union(u,v);
            }
            else
            {
                scanf("%d",&u);
                fin(u);//这里必须有   因为算法是延迟更新的    这里如果不查找   就不会更新
                printf("%d\n",under[u]);
            }
        }
    }
    return 0;
}
带权并查集

HDU3047 Zjnu Stadium

Posted on

最近一周好好把并查集和最短路走的尽量远一点 再然后就得往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;
}