POJ1988 Cube Stacking 题解

POJ1988 Cube Stacking 题解

原题链接：http://poj.org/problem?id=1988

题目大意

有$N$个方块，$P$个操作。一种操作是$M\ x\ y$表示把含$x$的方块堆移动到含$y$的方块堆顶部，另一种操作$C\ x$询问在含$x$方块的方块堆中，在$x$方块下面的方块数，然后输出。

解题思路

并查集的路径迭代。
定义$Up[i]$为$i$方块上的方块数（含自己），$Down[i]$为$i$方块下的方块数（不含自己），显然初始化$Up[i]=1,Down[i]=0$

对于每次$M$操作，我们可以发现含$x$的方块堆下多了含$y$的方块堆的方块数，即$Down[x]+=Up[y]$，含$y$的方块堆上多了含$x$的方块堆的方块数，即$Up[y]+=Down[x]$，但是两个方块堆中其他方块没有被修正，这里就要把$getfa(x)$这个函数做修改，每次$getfa(x)$的时候$Down[x]+=Down[fa[x]]$即可。
对于$C$操作，输出$Down[x]$就行了。

代码

#include<cstdio>
#include<algorithm>
using namespace std;
int N,Up[30005],Down[30005],fa[30005];
inline int getfa(int x){
    if (x==fa[x]) return x;
    int f=fa[x];
    fa[x]=getfa(fa[x]); //这里先修正Down[fa[x]]
    Down[x]+=Down[f]; //修正Down[x]
    return fa[x];
}
inline void merge(int x,int y){
    x=getfa(x);y=getfa(y);
    if (x==y) return ;
    Down[x]=Up[y];Up[y]+=Up[x];fa[x]=y;
}
inline char read(){
    char ch=getchar();
    while (ch!='M'&&ch!='C') ch=getchar();
    return ch;
}
int main()
{
    scanf("%d",&N);
    for (int i=1;i<=30000;i++) fa[i]=i,Up[i]=1,Down[i]=0; //初始化
    for (int i=1;i<=N;i++){
        char ch=read();
        if (ch=='M'){
            int x,y;
            scanf("%d%d",&x,&y);
            merge(x,y); //合并
        }
        else {
            int x;scanf("%d",&x);
            getfa(x); //计算前先修正
            printf("%d\n",Down[x]);
        }
    }
    return 0;
}