| Time Limit: 2000MS | Memory Limit: 30000K | |
| Total Submissions: 18900 | Accepted: 6568 | |
| Case Time Limit: 1000MS | ||
Description
Input
Output
Sample Input
6 M 1 6 C 1 M 2 4 M 2 6 C 3 C 4
Sample Output
1 0 2
Source
题目大意:
解题思路:有N个立方体和N个格子,1~N编号,一开始i立方体在i号格子上,每个格子刚好1个立方体。现在m组操作,M a b表示将a号立方体所在的格子的全部立方体放在b号立方体所在的格子的全部立方体上面。C x表示询问x号立方体下面的立方体的个数。
参考代码:在并查集的基础上,只需要知道x到父亲的距离以及父亲到底的距离就知道x到底的距离。sum[i]记录与根的距离,不断维护。
#include <iostream>
#include <vector>
#include <map>
using namespace std;
const int MAXN = 30010;
int p;
int father[MAXN], sum[MAXN], num[MAXN];
void init() {
for (int i = 1; i <= 30000; i++) {
father[i] = i;
sum[i] = 0;
num[i] = 1;
}
}
int find_set(int x) {
int tmp = father[x];
if (father[x] != x) {
father[x] = find_set(father[x]);
sum[x] += sum[tmp];
}
return father[x];
}
void union_set(int x, int y) {
x = find_set(x);
y = find_set(y);
if (x == y) return;
father[x] = y;
sum[x] += num[y];
num[y] += num[x];
}
void solve() {
for (int k = 0; k < p; k++) {
char op;
cin >> op;
if (op == 'M') {
int x, y, posx;
cin >> x >> y;
union_set(x, y);
} else if (op == 'C') {
int x, cnt = 0;
cin >> x;
find_set(x);
cout << sum[x] << endl;
}
}
}
int main() {
ios::sync_with_stdio(false);
while (cin >> p) {
init();
solve();
}
return 0;
}HDU 1988 Cube Stacking (数据结构-并查集),布布扣,bubuko.com
HDU 1988 Cube Stacking (数据结构-并查集)
原文:http://blog.csdn.net/wujysh/article/details/38358315