POJ1988 Cube Stacking

• 题目大意：有一些立方体，初始时每个立方体都在独立的一个栈中，共n个。支持两个操作，即把包含立方体的栈整体移动到包含立方体的栈顶（两栈合并）和询问立方体下方有多少个立方体。

• 思路：令栈顶元素为各个集合的代表元素，路径压缩的同时记录各节点至代表元素的距离（dist[ ]）以及以各元素为代表元素的元素数目(cnt[ ])。

• 代码如下：

#include<iostream>
#include<cstdio>
using namespace std;
const int maxn=30005;
int m,fa[maxn],cnt[maxn],dist[maxn];
int x,y;

int find(int x)
{
    if (fa[x]==x)
      return x;
    int fx=find(fa[x]);
    dist[x]+=dist[fa[x]];
    return fa[x]=fx;
}

void unite(int x,int y)
{
    int fx=find(x);
    int fy=find(y);
    fa[fy]=fx;
    dist[fy]=cnt[fx];
    cnt[fx]+=cnt[fy];
}

int main ()
{
    int x,y;
    char ch;
    scanf ("%d\n",&m);
    for (int i=1;i<=maxn;++i)
    {
        fa[i]=i;
        cnt[i]=1;
        dist[i]=0;
    }
    while (m--)
    {
        scanf ("%c",&ch);
        if (ch==‘M‘)
        {
            scanf ("%d%d\n",&x,&y);
            unite(x,y);
        }
        else
        {
            scanf("%d\n",&x);
            int fx=find(x);
            printf ("%d\n",cnt[fx]-dist[x]-1);
        }
    }
    return 0;
}