NC235745 拆路

题目链接

题目

题目描述

有 $n$ 个城镇，城镇之间有 $m$ 条道路相连，道路可以看成无向边。每一个城镇都有自己的一个繁荣度 $v_i$ ，一个城镇 $u$ 受到的影响 $p$ 是与$u$ 直接或者间接相连的所有城镇中，繁荣度的最大值。一个城镇 $u$ 与城镇 $v$ 是被视为直接或者间接相连的，当且仅当 $u=v$ 或者从 $u$ 出发，可以沿着某些道路到达 $v$ 。为了减少维护成本，现准备拆除其中的某一些路。具体来说，你需要维护以下两种操作：

1. ​ 'Q' $a$，询问 $a$ 城镇受到的影响 $p$ ；
2. ​ 'D' $a\ b$ ，删除 $a\ b$ 之间的道路。

输入描述

第一行输入两个整数 $n,m(1\le n,m \le 10^5)$ ，分别表示城镇的数量和道路的数量。第二行输入 $n$ 个整数 $v_1,v_2,...,v_n(1\le v_i \le 10^9)$ ，分别表示每一个城镇的繁荣度。接下来 $m$ 行，每行两个整数 $u,v(1\le u,v \le n)$ ，表示城镇 $u,v$ 之间有一条道路连接。保证不含有重边、自环。
接下来一行，输入一个整数 $Q(1\le Q \le 10^5)$ ，表示操作的个数。
接下来 $Q$ 行，每行描述一个操作，以'Q' $a(1\le a \le n)$ 或者'D' $a\ b(1\le a,b \le n)$ 的形式给出。对于删除操作，保证被删除的道路是存在的。

输出描述

对于每一个操作1，你都需要输出一个整数 $p$ ，表示城镇 $a$ 受到的影响。

示例1

输入

4 3
1 2 3 4
1 2
2 3
3 4
4
Q 1
D 2 3
Q 1
Q 3

输出

4
2
4

题解

知识点：并查集，离线。

普通并查集不支持撤销操作，这使得这道题难以解答，但转换一下思路，撤销的逆过程是新增，是并查集适用的操作，于是可以保存所有访问数据，离线逆向处理，再将答案保存最后输出即可。

时间复杂度 $O((m \log n + q)\log q + n)$

空间复杂度 $O(q+n+m)$

代码

#include <bits/stdc++.h>
 
using namespace std;
 
int w[100007];
int u[100007], v[100007];
int du[100007], dv[100007];
set<pair<int, int>> s;
int fa[100007];
int ans[100007];
 
int find(int x) {
    return fa[x] == x ? x : fa[x] = find(fa[x]);
}
 
void merge(int x, int y) {
    int rx = find(x);
    int ry = find(y);
    if (w[rx] < w[ry]) fa[rx] = ry;
    else fa[ry] = rx;
}
 
int main() {
    std::ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
    int n, m;
    cin >> n >> m;
    for (int i = 1;i <= n;i++) fa[i] = i;
    for (int i = 1;i <= n;i++) cin >> w[i];
    for (int i = 1;i <= m;i++) cin >> u[i] >> v[i];
    int q;
    cin >> q;
    for (int i = 1;i <= q;i++) {
        char op;
        cin >> op;
        if (op == 'Q') cin >> du[i];
        else if (op == 'D') {
            cin >> du[i] >> dv[i];
            s.insert({ du[i],dv[i] });
            s.insert({ dv[i],du[i] });
        }
    }
    for (int i = 1;i <= m;i++)
        if (s.find({ u[i],v[i] }) == s.end())
            merge(u[i], v[i]);
    for (int i = q;i >= 1;i--) {
        if (dv[i]) merge(du[i], dv[i]);
        else ans[i] = w[find(du[i])];
    }
    for (int i = 1;i <= q;i++) {
        if (ans[i]) cout << ans[i] << '\n';
    }
    return 0;
}