标签：rt P3402 ver int 题解 查集 tree rootfa rootdep

题目链接：https://www.luogu.com.cn/problem/P3402

题目大意：

给定 \(n\) 个集合，第 \(i\) 个集合内初始状态下只有一个数，为 \(i\)。

有 \(m\) 次操作。操作分为 \(3\) 种：

• 1 a b 合并 \(a,b\) 所在集合；

• 2 k 回到第 \(k\) 次操作（执行三种操作中的任意一种都记为一次操作）之后的状态；

• 3 a b 询问 \(a,b\) 是否属于同一集合，如果是则输出 \(1\) ，否则输出 \(0\)。

解题思路：

可持久化并查集。并查集 按秩合并，不要路径压缩。

示例程序：

#include <bits/stdc++.h>
using namespace std;
const int maxn = 100010;
struct Tree {
    int l, r,val;
} tree[maxn*80];
int tot, rootfa[maxn*2], rootdep[maxn*2];
int n, m;
void build(int l, int r, int& rt) {
    rt = ++tot;
    if (l == r) {
        tree[rt].val = l;
        return;
    }
    int mid = (l + r) / 2;
    build(l, mid, tree[rt].l);
    build(mid+1, r, tree[rt].r);
}
void update(int p, int val, int l, int r, int pt, int &rt) {
    tree[rt = ++tot] = tree[pt];
    if (l == r) {
        tree[rt].val = val;
        return;
    }
    int mid = (l + r) / 2;
    if (p <= mid)
        update(p, val, l, mid, tree[pt].l, tree[rt].l);
    else
        update(p, val, mid+1, r, tree[pt].r, tree[rt].r);
}
int query(int p, int l, int r, int rt) {
    if (l == r) return tree[rt].val;
    int mid = (l + r) / 2;
    if (p <= mid)
        return query(p, l, mid, tree[rt].l);
    else
        return query(p, mid+1, r, tree[rt].r);
}
int ffind(int ver, int x) {
    int fa = query(x, 1, n, rootfa[ver]);
    return fa == x ? x : ffind(ver, fa);
}
void funion(int ver, int x, int y) {
    x = ffind(ver-1, x);
    y = ffind(ver-1, y);
    if (x == y) {
        rootfa[ver] = rootfa[ver-1];
        rootdep[ver] = rootdep[ver-1];
    }
    else {
        int depx = query(x, 1, n, rootdep[ver-1]);
        int depy = query(y, 1, n, rootdep[ver-1]);
        if (depx > depy) swap(x, y);
        update(x, y, 1, n, rootfa[ver-1], rootfa[ver]);
        if (depx == depy) {
            update(y, depy+1, 1, n, rootdep[ver-1], rootdep[ver]);
        }
        else {
            rootdep[ver] = rootdep[ver-1];
        }
    }
}
int main() {
    ios::sync_with_stdio(0);
    cin >> n >> m;
    build(1, n, rootfa[0]);
    for (int i = 1; i <= m; i ++) {
        int op, k, a, b;
        cin >> op;
        if (op == 2) cin >> k;
        else cin >> a >> b;
        if (op == 1) {  // 合并a,b所在集合
            funion(i, a, b);
        }
        else if (op == 2) { // 回到第k次操作
            rootfa[i] = rootfa[k];
            rootdep[i] = rootdep[k];
        }
        else {  // 询问a,b是否属于同一集合
            rootfa[i] = rootfa[i-1];
            rootdep[i] = rootdep[i-1];
            cout << (ffind(i, a) == ffind(i, b) ? 1 : 0) << endl;
        }
    }
    return 0;
}

标签：rt,P3402,ver,int,题解,查集,tree,rootfa,rootdep
来源： https://www.cnblogs.com/quanjun/p/15523414.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。