洛谷P1197 星球大战【并查集】

题目：https://www.luogu.org/problemnew/show/P1197

题意：有n个结点m条无向边，k次操作每次摧毁一个结点并询问此时有多少连通块。

思路：平时在线的搞多了都没想到这道题完全可以存下结果之后输出。

对于那些要被摧毁的城市，我们只需要先都摧毁，然后倒序的进行恢复。就可以求出每一次操作的结果了。

因为对于摧毁我们不好用并查集，但是对于重建就比较好用并查集了。

所以分成两步，把所有没有被摧毁的连通的结点都合并起来。

然后按照倒序加入被摧毁的结点，将他和他所邻接的结点进行合并。

#include<cstdio>
#include<cstdlib>
#include<map>
#include<set>
#include<cstring>
#include<algorithm>
#include<vector>
#include<cmath> 
#include<stack>
#include<queue>
#include<iostream>

#define inf 0x7fffffff
using namespace std;
typedef long long LL;
typedef pair<string, string> pr;

int n, m, k;
const int maxn = 4e5 + 5;
int par[maxn];
int head[maxn];
struct edge{
    int u, v, nxt;
}e[maxn];
int tot = 0;
bool destroy[maxn];

void addedge(int u, int v)
{
    e[tot].u = u;
    e[tot].v = v;
    e[tot].nxt = head[u];
    head[u] = tot++;
    e[tot].u = v;
    e[tot].v = u;
    e[tot].nxt = head[v];
    head[v] = tot++;
}

int ffind(int x)
{
    if(par[x] == x)return x;
    else return par[x] = ffind(par[x]);
}

void init()
{
    for(int i = 0; i < n; i++){
        par[i] = i;
        head[i] = -1;
        destroy[i] = false;
    }
}

stack<int>sss;
stack<int>ans;
int main()
{
    scanf("%d%d", &n, &m);
    init(); 
    for(int i = 0; i < m; i++){
        int u, v;
        scanf("%d%d", &u, &v);
        addedge(u, v);
    } 
    scanf("%d", &k);
    for(int i = 0; i < k; i++){
        int d;
        scanf("%d", &d);
        destroy[d] = true;
        sss.push(d);
    }
    
    int cnt = n - k;
    for(int i = 0; i < tot; i++){
        if(!destroy[e[i].u] && !destroy[e[i].v]){
            int fu = ffind(e[i].u), fv = ffind(e[i].v);
            if(fu != fv){
                cnt--;
                par[fu] = fv;
            }
        }
    }
    ans.push(cnt);
    while(!sss.empty()){
        int x = sss.top();sss.pop();
        cnt++;
        destroy[x] = false;
        for(int i = head[x]; i != -1; i = e[i].nxt){
            if(!destroy[e[i].u] && !destroy[e[i].v]){
                int fu = ffind(e[i].u), fv = ffind(e[i].v);
                if(fu != fv){
                    cnt--;
                    par[fu] = fv;
                }
            }
        }
        ans.push(cnt);
    }
    
    while(!ans.empty()){
        printf("%d
", ans.top());
        ans.pop();
    }
    
    return 0; 
}