叠积木(加权并查集)

叠积木

题目描述:
约翰和贝西在叠积木。共有30000块积木,编号为1到30000。一开始,这些积木放在地上,自然地分成N堆。贝西接受约翰的指示,把一些积木叠在另一些积木的上面。一旦两块积木相叠, 彼此就再也不会分开了,所以最后叠在一起的积木会越来越高。约翰让贝西依次执行P条操作,操作分为两种:
第一种是移动操作,格式为“移动X到Y的上面”。X和Y代表两块积木的编号,意思是将X所的那堆积木,整体叠放到Y所在的那堆积木之上;
第二种是统计操作,格式为“统计Z下方的积木数量”。Z代表一块积木的编号,意思是贝西需要报告在编号为Z的积木之下还有多少块积木
请编写一个程序,帮助贝西回答每条统计问题。
输入输出格式
输入格式:
第一行:单个整数:P,1≤P≤10^5
第二行到第P + 1行:每行描述一条命令,如果这行开头的字母是 M,代表一条移动命令,后面的两个整数代表上文中的X和Y;如果开头字母是 C,代表一条统计命令。后面的整数代表上文中的Z,保证所有的移动命令都有意义,X和Y不会已经出现在同一堆积木里
输出格式:
对每一个统计命令,输出正确回答,用换行符分开每个查询的结果
输入输出样例
输入样例:
6
M 1 6
C 1
M 2 4
M 2 6
C 3
C 4
输出样例:
1
0
2
思路
数组top[x]表示x所属的栈顶元素。
数组cnt[x]表示x到栈底的元素个数。
数组father[x]表示x的栈底元素。

#include<iostream>
using namespace std;
const int maxn=30010;
int n,father[maxn],cnt[maxn],top[maxn];
char c;
int find(int x)
{
    if(father[x]==x)
    return x;
    int fa=father[x];
    father[x]=find(fa);
    cnt[x]+=cnt[fa];
    top[x]=top[fa];
    return father[x];
}
void unionn(int f1,int f2)
{
    father[f1]=f2;
    find(top[f2]);
    cnt[f1]=cnt[top[f2]]+1;
    top[f2]=top[f1];
}
int main()
{
    int x,y;
    cin>>n;
    for(int i=1;i<=maxn;i++)
    father[i]=top[i]=i;
    for(int i=1;i<=n;i++)
    {
        cin>>c;
        if(c=='M')
        {
            cin>>x>>y;
            int f1=find(x);
            int f2=find(y);
            if(f1!=f2)
            unionn(f1,f2);
        }
        else
        {
            cin>>x;
            find(x);
            cout<<cnt[x]<<endl;
        }
    }
    return 0;
}
原文地址:https://www.cnblogs.com/cax1165/p/6070997.html