洛谷 P1197 [JSOI2008]星球大战

题目：

https://www.luogu.org/problem/show?pid=1197

题目描述

很久以前，在一个遥远的星系，一个黑暗的帝国靠着它的超级武器统治者整个星系。某一天，凭着一个偶然的机遇，一支反抗军摧毁了帝国的超级武器，并攻下了星系中几乎所有的星球。这些星球通过特殊的以太隧道互相直接或间接地连接。

但好景不长，很快帝国又重新造出了他的超级武器。凭借这超级武器的力量，帝国开始有计划地摧毁反抗军占领的星球。由于星球的不断被摧毁，两个星球之间的通讯通道也开始不可靠起来。现在，反抗军首领交给你一个任务：给出原来两个星球之间的以太隧道连通情况以及帝国打击的星球顺序，以尽量快的速度求出每一次打击之后反抗军占据的星球的连通快的个数。（如果两个星球可以通过现存的以太通道直接或间接地连通，则这两个星球在同一个连通块中）。

输入输出格式

输入格式：
输入文件第一行包含两个整数，N (1 <= N <= 2M) 和M (1 <= M <= 200,000)，分别表示星球的数目和以太隧道的数目。星球用0~N-1的整数编号。

接下来的M行，每行包括两个整数X, Y，其中（0<=X<> Y < N），表示星球X和星球Y之间有以太隧道。注意所有的以太隧道都是双向的。

接下来一行是一个整数K，表示帝国计划打击的星球个数。

接下来的K行每行一个整数X，满足0<=X < N，表示帝国计划打击的星球编号。帝国总是按输入的顺序依次摧毁星球的。
输出格式：
输出文件的第一行是开始时星球的连通块个数。

接下来的K行，每行一个整数，表示经过该次打击后现存星球的连通块个数。

输入输出样例

输入样例#1：
8 13
0 1
1 6
6 5
5 0
0 6
1 2
2 3
3 4
4 5
7 1
7 2
7 6
3 6
5
1
6
3
5
7
输出样例#1：
1
1
1
2
3
3

题解：

用并查集维护星球之间的联通状况
如果按照题目要求实现，需要支持删除某点，维护森林，显然通过并查集是较难实现的。
考虑倒序实现，将题目倒转思考，正序删除某点，即为逆序地增加某点，同时维护森林，用并查集实现；

注意：

要求输出的是打击 后！现存的！联通块数量。
所以在倒序加点时，要先把联通块的数目+1，表示新增一个孤立元素，然后再进行合并操作；

代码：

#include<iostream>
#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;
const int N=400000+50,M=200000+50;
bool bo[N];
int s[M],t[M],ans[N],fa[N],a[N];
vector<int> mp[N];
int n,m,k,cnt;
int find(int x){
    if(x==fa[x]) return x;
    else return fa[x]=find(fa[x]);
}
void merge(int a,int b){
    int fs=find(a),ft=find(b);
    if(fs==ft) return ;
    cnt--;
    fa[fs]=ft;
    return ;
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++){
        scanf("%d%d",&s[i],&t[i]);
    }

    for(int i=0;i<=n;i++) fa[i]=i;
    scanf("%d",&k);
    cnt=n-k;
    for(int i=1;i<=k;i++) scanf("%d",&a[i]),bo[a[i]]=1;
    for(int i=1;i<=m;i++){
        if(!bo[s[i]]&&!bo[t[i]]) {
            merge(s[i],t[i]);
        }
        else {
            if(bo[s[i]])
                mp[s[i]].push_back(t[i]);
            if(bo[t[i]])
                mp[t[i]].push_back(s[i]);
        }
    }

    ans[k]=cnt;

    for(int i=k;i>=1;i--){
        bo[a[i]]=0;
        cnt++;
        int sz=mp[a[i]].size();
        for(int j=0;j<sz;j++){
            int to=mp[a[i]][j];
            if(!bo[to])
            merge(a[i],to);
        }
        ans[i-1]=cnt;
    }
    printf("%d\n",cnt);
    for(int i=1;i<=k;i++) printf("%d\n",ans[i]);
    return 0;
}