uva 11987 Almost Union-Find(并查集的删除操作)


题意:

初始时,一共有n个元素的组合1,2,3....n

给出三个操作

1 p q:合并p,q所在的集合

2 p q:把p移动到q所在的集合

3 p:输出p所在的集合的元素的个数


思路:1,3相当简单,赤裸裸的并查集就好了 
麻烦的是2,并查集是单向的,只知道父亲不知道儿子 
所有如果i是叶子节点无所谓 
如果是父亲节点就会十分的蛋疼: 
你是到另一个集合里面去了,你的儿子们呢? 
把i这个点复制一下(编号为cnt++),当作叶子节点扔到新集合里去 
原来的i仅仅有作为集合编号的作用,可以认为是个空点 ,只是为了不让原来的并查集出现错误,防止他的儿子们找不到根节点

重新找一个节点,记录这个节点自己的信息,也就是sum = 他自己, num = 1;

注意:合并或者查询等的时候一定要合并id[i],因为原来的i只是一个空节点了,为了保证并查集不出错而存在的,在del函数的时候,传参

要穿本身,因为一会要用到本身,新的节点的sum是他本身,并且对id数组更新的时候,下标也是他本身


代码:

#include<bits/stdc++.h>
using namespace std;
const int maxn = 3*1e5+5;
int pre[maxn], id[maxn], num[maxn], sum[maxn], cnt;

void init(int n)
{
    cnt = n;
    for(int i = 1; i <= n; i++)
        pre[i] = i, num[i] = 1, sum[i] = i, id[i] = i;
}

int Find(int x)
{
    int r = x;
    while(pre[r] != r) r = pre[r];
    int i = x, j;
    while(pre[i] != r)
    {
        j = pre[i];
        pre[i] = j;
        i = j;
    }
    return r;
}

void join(int x, int y)
{
    int a = Find(x);
    int b = Find(y);
    if(a != b)
    {
        pre[b] = a;
        sum[a] += sum[b];
        num[a] += num[b];
    }
}

void del(int x)
{
    int r = Find(id[x]);
    num[r]--;
    sum[r] -= x;
    cnt++;
    pre[cnt] = cnt;
    num[cnt] = 1;
    sum[cnt] = x;
    id[x] = cnt;
}

int main(void)
{
    int n, m;
    while(cin >> n >> m)
    {
        init(n);
        while(m--)
        {
            int cmd, x, y;
            scanf("%d", &cmd);
            if(cmd == 1)
            {
                scanf("%d%d", &x, &y);
                join(id[x], id[y]);
            }
            if(cmd == 2)
            {
                scanf("%d%d", &x, &y);
                 //注意要判断下是否在同一颗树上
                if(Find(id[x]) != Find(id[y]))
                {
                    del(x);
                    join(id[x], id[y]);
                }
            }
            if(cmd == 3)
            {
                scanf("%d", &x);
                int r = Find(id[x]);
                printf("%d %d\n", num[r], sum[r]);
            }
        }
    }
    return 0;
}


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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值