首页 > 其他 > 详细

uva11987

时间:2017-11-08 20:41:12      阅读:198      评论:0      收藏:0      [点我收藏+]

题意是实现一个带删除功能的并查集。

这题的做法是,比如你要删除x,你就相当于把x剥离出来,开一个新的点去记录新的x,同时把原来x的父节点fa[x]做关于删除x节点信息的操作。

#include <cstdio>

using namespace std;

const int maxn = 1000000 + 5;

int n, m, tot;

int id[maxn * 2], fa[maxn], cnt[maxn], sum[maxn];

int find(int x)
{
    return fa[x] == x ? x : fa[x] = find(fa[x]);
}

void Union(int x, int y)
{
    int fx = find(x), fy = find(y);
    if (fx != fy)
    {
        fa[fx] = fy;
        cnt[fy] += cnt[fx];
        sum[fy] += sum[fx];
    }
}

void del(int x)
{
    int fx = find(id[x]);
    sum[fx] -= x;
    cnt[fx]--;
    tot++;
    id[x] = tot;
    fa[tot] = tot;
    sum[tot] = x;
    cnt[tot] = 1;
}

int main()
{
//    freopen("uva11987.in","r",stdin);
    while (~scanf("%d%d", &n, &m))
    {
        tot = n;
        for (int i = 1; i <= n; i++)
        {
            id[i] = fa[i] = sum[i] = i;
            cnt[i] = 1;
        }
        for (int i = 1; i <= m; i++)
        {
            int op;
            scanf("%d", &op);
            if (op == 1) 
            {
                int x, y;
                scanf("%d%d", &x, &y);
                Union(id[x], id[y]);
            }
            else
            if (op == 2)
            {
                int x, y;
                scanf("%d%d", &x, &y);
                int fx = find(id[x]);
                int fy = find(id[y]);
                if (fx != fy)
                {
                    del(x);
                    Union(id[x], id[y]);
                }
            }
            else
            if (op == 3)
            {
                int x;
                scanf("%d", &x);
                int fx = find(id[x]);
                printf("%d %d\n", cnt[fx], sum[fx]);
            }
        } 
    }
    return 0;
}

 

uva11987

原文:http://www.cnblogs.com/yohanlong/p/7805947.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!