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
#include
#include
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]);
            }
        }
    }
}

 

你可能感兴趣的:(并查集,并查集)