uva-11987 Almost Union-Find (并查集)

题意:

     初始给定n个集合:{1},{2},…,{n},要求支持三种操作:

      "1 p q":若p,q不在同一集合,将它们所在的集合合并成一个

      "2 p q":若p,q不在同一集合,将元素p移动到q所在的集合

      "3 p"   :询问p所在集合的元素个数及元素和

       总共m个操作,1<=n,m<=10^5

分析:第一种操作直接合并就可以,第三种操作则开俩个数组,一个记录集合的个数,另一个记录集合的和;

而操作二因为要删除元素,但删除元素对原集合的结构造成影响,所以我们开一个数组来标记他的编号。

删除p元素,改变p的编号,所以原编号p就没有用了,拿新的编号p直接合并就行
 

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
typedef long long ll;
const int INF=0x3f3f3f3f;
const int N=200000;
int f[N+10],sum[N+10],Size[N+10],p[N+10];
void init()
{
    for(int i=0; i<=N; i++)
    {
        f[i]=p[i]=sum[i]=i;
        Size[i]=1;
    }
}
int getf(int v)
{
    return f[v]==v?v:f[v]=getf(f[v]);
}
int main()
{
    int n,m;
    while(~scanf("%d%d",&n,&m))
    {
        init();
        int t=n;
        while(m--)
        {
            int a,b,c;
            scanf("%d",&a);
            if(a==1)
            {
                scanf("%d%d",&b,&c);
                int t1=getf(p[b]);
                int t2=getf(p[c]);
                if(t1!=t2)
                {
                    sum[t1]+=sum[t2];
                    Size[t1]+=Size[t2];
                    f[t2]=t1;
                }
            }
            else if(a==2)
            {
                scanf("%d%d",&b,&c);
                int t1=getf(p[b]);
                int t2=getf(p[c]);
                if(t1!=t2)
                {
                    sum[t1]-=b;
                    Size[t1]--;
                    t1=p[b]=++t;
                    f[t1]=t2;
                    sum[t2]+=b;
                    Size[t2]++;
                }
            }
            else if(a==3)
            {
                scanf("%d",&b);
                int T=getf(p[b]);
                printf("%d %d\n",Size[T],sum[T]);
            }
        }
    }
}

 

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值