洛谷P2342-叠积木

Problem 洛谷P2342-叠积木

Accept: 373   Submit: 1.1k

Time Limit: 1000 mSec    Memory Limit : 128MB

Problem Description

约翰和贝西在叠积木。共有30000块积木，编号为1到30000。一开始，这些积木放在地上，自然地分成N堆。贝西接受约翰的指示，把一些积木叠在另一些积木的上面。一旦两块积木相叠， 彼此就再也不会分开了，所以最后叠在一起的积木会越来越高。约翰让贝西依次执行P条操作，操作分为两种：

1. 1. 　　 第一种是移动操作，格式为“移动X到Y的上面”。X和Y代表两块积木的编号，意思是将X所的那堆积木，整体叠放到Y所在的那堆积木之上；
   2. 　     第二种是统计操作，格式为“统计Z下方的积木数量”。Z代表一块积木的编号，意思是贝西需要报告在编号为Z的积木之下还有多少块积木；

请编写一个程序，帮助贝西回答每条统计问题。

 Input

第一行：单个整数：P，1 ≤ P ≤ 10^5

第二行到第P + 1行：每行描述一条命令，如果这行开头的字母是 M，代表一条移动命令，后面的两个整数代表上文中的X和Y；如果开头字母是 C，代表一条统计命令。后面的整数代表上文中的Z，保证所有的移动命令都有意义，X和Y不会已经出现在同一堆积木里

 Output

　　对每一个统计命令，输出正确回答，用换行符分开每个查询的结果

 Sample Input

6
M 1 6
C 1
M 2 4
M 2 6
C 3
C 4

Sample output

1
0
2

题目链接：https://www.luogu.org/problemnew/show/P2342

题解：感觉这个题和一般的带权并查集有那么一点点区别，记录下来。

每个集合的父节点是一堆积木的最下面的积木，权值是当前积木到所在集合父节点的积木个数。

两点需要注意的，一是集合内部关系更改，也就是rel数组，这个是在路径压缩的过程中递归实现的，二是集合之间关系的更改，不仅需要rel数组，还要利用Size数组（记录每个集合当前的大小），当把一个集合放在另一个集合上面时，上面积木的底部到下面积木的底部的积木个数就是Size[下面积木]。

别忘了当一堆积木放在别的积木上面的时候，Size要修改为0。

 #include <iostream>
 #include <cstdio>
 #include <cstring>
 #include <cstdlib>
 using namespace std;

 const int maxn = +;

 int rel[maxn],pre[maxn];
 int Size[maxn];

 int findn(int x){
     if(x == pre[x]) return x;
     int temp = pre[x];
     pre[x] = findn(pre[x]);
     rel[x] += (rel[temp]);
     return pre[x];
 }

 void merge_node(int x,int y){
     int fx = findn(x);
     int fy = findn(y);
     if(fx != fy){
         pre[fx] = fy;
         rel[fx] = Size[fy];
         Size[fy] += Size[fx];
         Size[fx] = ;
     }
 }

 int main()
 {
     //freopen("input.txt","r",stdin);
     int p,n,x,y;
     char ch[];
     n = ;
     for(int i = ;i < n;i++){
         pre[i] = i;
         rel[i] = ;
         Size[i] = ;
     }
     scanf("%d",&p);
     for(int i = ;i <= p;i++){
         scanf("%s",ch);
         if(ch[] == 'M'){
             scanf("%d%d",&x,&y);
             merge_node(x,y);
         }
         else{
             scanf("%d",&x);
             findn(x);
             printf("%d\n",rel[x]);
         }
     }
     return ;
 }