Cube Stacking————北大OJ

题目链接
http://poj.org/problem?id=1611
思路
题意说是有n块砖,编号从1到n,有两种操作,操作一(‘M’)是把含有x编号的那一堆砖放到含有编号y的那一堆砖的上面,操作二(‘C’)是查询编号为x的砖的下面有多少块砖。传统并查集+维护size和deep两个数组,find函数在路径压缩的过程中需要把深度也依次加上,到代表这里的时候就是这个集合的最大深度,合并过程中要先将被合并的个数加到合并别人的集合的深度上去,然后在合并个数,最后将被合并的集合清零。
c++代码

#include<iostream>
#include<algorithm>
#include<stdio.h>
const int N =1e5+5;
using namespace std;
int deep[N];
int size[N];
int p[N];
int find(int x)
{
    
    
    if(p[x]!=x) 
    {
    
    
        int t=p[x];
        p[x]=find(p[x]);
        deep[x]+=deep[t];
    }
    return p[x];
}
void Union(int x,int y)
{
    
    
    int xx=find(x);
    int yy=find(y);
    if(xx!=yy)
    {
    
    
        p[xx]=yy;
        deep[xx]+=size[yy];
        size[yy]+=size[xx];
        size[xx]=0;
    }
}
int main()
{
    
    
	int n; 
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
    
    
        p[i]=i;
        size[i]=1;
    }
    for(int i=1;i<=n;i++)
    {
    
    
        char op[5];
        scanf("%s",op);
        if(op[0]=='M')
        {
    
    
            int a,b;
            scanf("%d %d",&a,&b);
            Union(a,b);
        }
        else if(op[0]=='C')
        {
    
    
            int a;
            scanf("%d",&a);
            find(a);
            printf("%d\n",deep[a]);
        }
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/Star_Platinum14/article/details/111937795