【数据结构 并查集 主席树】洛谷_3402 可持久化并查集

题意

顾名思义，写出一个支持可持久化的并查集。

思路

用可持久化数据结构记录每个版本时每个点的父亲，然后我觉得就没什么好讲的了。

不过，在此用路径压缩会炸空间，因为每次路径压缩会新增很多节点，所以我们采用按秩合并来优化查询时间。

所谓按秩合并，就是通过特殊方法合并，你可以把“秩”视作集合大小，或树的深度，当两个集合合并时，令查询代价小的连向查询代价大的，就只会增加少量的查询代价。

还有很多关于可持久化、以及空间之类的细节。

代码

#include<cstdio>
#include<algorithm>

struct SegmentTree {
    int lc, rc;
    int fa, dep;
} tree[6000001];
int root[200001];
int tot, n, m;

int build(int l, int r) {
    int p = ++tot;
    if (l == r) {
        tree[p].fa = l;
        return p;
    }
    int mid = l + r >> 1;
    tree[p].lc = build(l, mid);
    tree[p].rc = build(mid + 1, r);
    return p;
}

int change(int now, int l, int r, int pos, int fa) {
    int p = ++tot;
    tree[p] = tree[now];
    if (l == r) {
        tree[p].fa = fa;
        return p;
    }
    int mid = l + r >> 1;
    if (pos <= mid) tree[p].lc = change(tree[now].lc, l, mid, pos, fa);
    else tree[p].rc = change(tree[now].rc, mid + 1, r, pos, fa);
    return p;
}

int ask(int now, int l, int r, int pos) {
    if (l == r) return now;
    int mid = l + r >> 1;
    if (pos <= mid) return ask(tree[now].lc, l, mid, pos);
    else return ask(tree[now].rc, mid + 1, r, pos);
}

int find(int ver, int x) {
    int fa = ask(ver, 1, n, x);
    if (tree[fa].fa == x) return fa;
    return find(ver, tree[fa].fa);
}

void add(int now, int l, int r, int pos) {
    if (l == r) {
        tree[now].dep++;
        return;
    }
    int mid = l + r >> 1;
    if (pos <= mid) add(tree[now].lc, l, mid, pos);
    else add(tree[now].rc, mid + 1, r, pos);
}

void connect(int ver, int a, int b) {
    root[ver] = root[ver - 1];
    int f1 = find(root[ver], a), f2 = find(root[ver], b);
    if (tree[f1].fa == tree[f2].fa) return;
    if (tree[f1].dep > tree[f2].dep) std::swap(f1, f2);
    root[ver] = change(root[ver - 1], 1, n, tree[f1].fa, tree[f2].fa);
    if (tree[f1].dep == tree[f2].dep) add(root[ver], 1, n, tree[f2].fa);//注意此
}

void check(int ver, int a, int b) {
    root[ver] = root[ver - 1];
    int f1 = find(root[ver], a), f2 = find(root[ver], b);
    if (tree[f1].fa == tree[f2].fa) printf("1\n");
    else printf("0\n");
}

int main() {
    scanf("%d %d", &n, &m);
    int op, a, b, k;
    root[0] = build(1, n);
    for (int i = 1; i <= m; i++) {
        scanf("%d", &op);
        switch (op) {
            case 1:scanf("%d %d", &a, &b), connect(i, a, b); break;
            case 2:scanf("%d", &k), root[i] = root[k]; break;
            case 3:scanf("%d %d", &a, &b), check(i, a, b); break;
        }
    }
}