num数组表示某个元素所在组有的元素数目,刚开始在压缩路径时候我还把这一堆的所有元素的num全改变成根节点的num,后来发现并不用这样。above表示某个元素下边有多少的元素,这个数组需要在压缩路径时候维护。
#include <iostream> #include <cstdio> using namespace std; const int MAX_N = 30000 + 100; int num[MAX_N], p[MAX_N], under[MAX_N]; int _find(int x) { if(p[x] != x) { int root = _find(p[x]); under[x] += under[p[x]]; //num[x] = num[p[x]]; return p[x] = root; } return x; } int main() { int P; scanf("%d", &P); for(int i = 0; i < MAX_N; i++) p[i] = i, under[i] = 0, num[i] = 1; char str[3]; int a, b, u, v; for(int i = 0; i < P; i++) { scanf("%s", str); if(str[0] == 'M') { scanf("%d%d", &a, &b); u = _find(a); v = _find(b); if(u != v) { p[u] = v; under[u] += num[v]; num[v] += num[u]; } } else { scanf("%d", &a); _find(a); printf("%d\n", under[a]); } } return 0; }
本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系我们删除。