HDU 2818 Building Block

---

题目连接 http://acm.hdu.edu.cn/showproblem.php?pid=2818

题意:给定N个blocks，分在N个堆里，然后又P个操作，每次将x所在的堆放在y所在的堆上，或者询问x的下面有几个blocks

做法：带权并查集

因为要查询x的下面有多少blocks，所以我们在普通并查集的基础上在维护两个域size[x]和under[x]，分别表示x所在堆的大小以及x下面的元素。

在合并的时候，我们分别取x,y的堆的最下面一块，也就是他们的根a,b.a和b相等就不用处理了。如果不相等，那么就让F[a] = b.而在这之前，我们要维护size和under，所有x原来所在的堆的每个元素的under都要增加size[b],如果全都修改会超时，所以我们之修改under[a],把其它修改放在压缩里面，要查哪一个再更新。同时，为了方便我们只把size存在根上，也就是size[b]+=size[a],size[a] = 0。

在find的时候，我们进行压缩，这时候更新under[x],under[x]+=under[fx]就可以了。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <vector>
#include <set>
#include <queue>
#include <set>
#include <map>
#include <cstring>
#include <functional>
#include <cmath>
typedef long long ll;
using namespace std;
const int MAXN = 30010;
int F[MAXN],size[MAXN],under[MAXN];
int f(int x){if(F[x] == x)return x;int fx = F[x];F[x] = f(F[x]);under[x]+=under[fx];return F[x];
}
void uni(int x,int y){int a = f(x),b = f(y);if(a == b)return ;under[a] += size[b];size[b]+=size[a];size[a] = 0;F[a] = b;
}
int main(){ios::sync_with_stdio(0);for(int i=0;i<MAXN;i++){F[i] = i;size[i] = 1;}int p;cin>>p;while(p--){char cmd;cin>>cmd;if(cmd == 'M'){int x,y;cin>>x>>y;uni(x,y);}else{int x;cin>>x;f(x);cout<< under[x] <<endl;}}return 0;
}