洛谷__P1197 [JSOI2008] 星球大战（并查集）

题目链接：P1197 [JSOI2008] 星球大战 - 洛谷

---

题目大意:

给定无向图，按顺序删除点（及相连边），

求每次删除后的连通块数量。

---

思路：

逆向思维，把删除转换成逆向修建；

标记要删除的点，按顺序从后往前恢复，

用并查集维护之间的关系即可；

---

代码：

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<vector>
#include<queue>
#include<deque>
#include<stack>
#include<set>
#include<map>
#include<unordered_set>
#include<unordered_map>
#include<bitset>
#include<tuple>
#define inf 72340172838076673
#define int long long
#define endl '\n'
#define F first
#define S second
#define  mst(a,x) memset(a,x,sizeof (a))
using namespace std;
typedef pair<int, int> pii;

const int N = 400086, mod = 998244353;

int n, m, k;
int h[N], ne[N], e[N], idx;
int f[N];
int bk[N];
bool ok[N];

void add(int a, int b) {
    e[idx] = b;
    ne[idx] = h[a];
    h[a] = idx++;
}

int find(int u) {
    if (f[u] == u) return u;
    return f[u] = find(f[u]);
}

void get(int u, int v) {
    u = find(u), v = find(v);
    if (u != v) f[u] = v;
}

void solve() {

    mst(h, -1);
    cin >> n >> m;
    
    for (int i = 0; i <= n; i++) f[i] = i;
    
    for (int i = 1; i <= m; i++) {
        int a, b;
        cin >> a >> b;
        add(a, b), add(b, a);
    }
    
    cin >> k;
    for (int i = 1; i <= k; i++) {
        cin >> bk[i];
        ok[bk[i]] = true;//标记要删除的点
    }
    
    int res = n - k;//先把剩下没删的点看成是独立的块
    for (int i = 0; i < idx; i += 2) {
        if (!ok[e[i]] && !ok[e[i + 1]] && find(e[i]) != find(e[i + 1])) {
            res--;
            get(e[i], e[i + 1]);
        }
    }
    
    stack<int> st;
    st.push(res);
    
    while (k--) {//从后往前遍历
        int u = bk[k + 1];
        ok[u] = false;//恢复
        res++;//总连通块数增加
        
        for (int i = h[u]; ~i; i = ne[i]) {
            int j = e[i];
            if (!ok[j] && find(u) != find(j)) {
                res--;//能合并的话 总连通块数减少1
                get(u, j);
            }          
        }
        
        st.push(res);
    }
    
    
    while (st.size()) {
        cout << st.top() << endl;
        st.pop();
    }
    
}

signed main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr), cout.tie(nullptr);
    
    int T = 1;
// cin >> T;
    while (T--) solve();
    
    return 0;
}