题意:有n个方块栈,一开始每个栈里只有一个方块,方块编号为栈编号,两种操作,一种是把方块a所在的栈全放在方块b所在的栈上面,另一种要求输出方块a的下面有多少个方块。
解法:并查集。用数组size[i]维护以i为根的集合大小,数组ans[i]维护方块i下面有多少方块,那么当两个栈合并时,栈a的底的ans即为栈b的size,而两个栈合并后根还是栈b的栈底,所以栈b的size+=栈a的size;当查询一个方块a下面有多少方块时,ans[a]+=ans[father[a]]。
代码:
#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<string>
#include<string.h>
#include<math.h>
#include<limits.h>
#include<time.h>
#include<stdlib.h>
#include<map>
#include<queue>
#include<set>
#include<stack>
#include<vector>
#include<iomanip>
#define LL long long
#define lson l, m, rt << 1
#define rson m + 1, r, rt << 1 | 1using namespace std;int size[30005], ans[30005], father[30005];
int Find(int a)
{if(father[a] != a){int fa = father[a];father[a] = Find(father[a]);ans[a] += ans[fa];}return father[a];
}
void Union(int a, int b)
{int fa = Find(a), fb = Find(b);ans[fa] = size[fb];size[fb] += size[fa];father[fa] = fb;
}
int main()
{int q;scanf("%d", &q);for(int i = 0; i <= 30000; i++) size[i] = 1;for(int i = 0; i <= 30000; i++) father[i] = i;memset(ans, 0, sizeof ans);while(q--){char op[2];int a, b;scanf("%s%d", op, &a);if(op[0] == 'M'){scanf("%d", &b);Union(a, b);}else{Find(a);printf("%d\n", ans[a]);}}return 0;
}