UVA_11987
操作2实际上相当于先把x删掉然后再和y合并,因此和普通的并查集相比只是多了一个删除某个点的操作。
对于删除操作,是不能随便改变原来树中p[]的指向的,比较容易能够举出反例。其实我们只是想在不影响其他元素的前提下把x删掉,但不如换个思路,将x这个点对它所在的集合的“影响”变为0。然后再新开一个节点表示x,这样就相当于把x从原来的集合中剥离出来了。因此相比原来的并查集,需要多一个id[x]数组表示点x现在的标号,有了上面的思路其余的操作就不难写出来了。
#include<stdio.h> #include<string.h> #define MAXD 200010 int N, M, p[MAXD], id[MAXD], num[MAXD], cnt; long long sum[MAXD]; void init() { int i; for(i = 1; i <= N; i ++) id[i] = p[i] = sum[i] = i, num[i] = 1; cnt = N; } int find(int x) { return p[x] == x ? x : (p[x] = find(p[x])); } void Union(int x, int y) { int tx = find(id[x]), ty = find(id[y]); p[tx] = ty, num[ty] += num[tx], sum[ty] += sum[tx]; } void Delete(int x) { int tx = find(id[x]); -- num[tx], sum[tx] -= x; id[x] = ++ cnt, p[id[x]] = id[x], num[id[x]] = 1, sum[id[x]] = x; } void solve() { int i, x, y, op; for(i = 0; i < M; i ++) { scanf("%d", &op); if(op == 1) { scanf("%d%d", &x, &y); if(find(id[x]) != find(id[y])) Union(x, y); } else if(op == 2) { scanf("%d%d", &x, &y); if(find(id[x]) != find(id[y])) Delete(x), Union(x, y); } else { scanf("%d", &x); printf("%d %lld\n", num[find(id[x])], sum[find(id[x])]); } } } int main() { while(scanf("%d%d", &N, &M) == 2) { init(); solve(); } return 0; }