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;
}

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

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值