BZOJ 1015 星球大战 离线并查集 好题

Description

　　很久以前，在一个遥远的星系，一个黑暗的帝国靠着它的超级武器统治者整个星系。某一天，凭着一个偶然的
机遇，一支反抗军摧毁了帝国的超级武器，并攻下了星系中几乎所有的星球。这些星球通过特殊的以太隧道互相直
接或间接地连接。 但好景不长，很快帝国又重新造出了他的超级武器。凭借这超级武器的力量，帝国开始有计划
地摧毁反抗军占领的星球。由于星球的不断被摧毁，两个星球之间的通讯通道也开始不可靠起来。现在，反抗军首
领交给你一个任务：给出原来两个星球之间的以太隧道连通情况以及帝国打击的星球顺序，以尽量快的速度求出每
一次打击之后反抗军占据的星球的连通快的个数。（如果两个星球可以通过现存的以太通道直接或间接地连通，则
这两个星球在同一个连通块中）。
Input

　　输入文件第一行包含两个整数，N (1 < = N < = 2M) 和M (1 < = M < = 200,000)，分别表示星球的
数目和以太隧道的数目。星球用 0 ~ N-1的整数编号。接下来的M行，每行包括两个整数X, Y，其中（0 < = X <>
Y 表示星球x和星球y之间有“以太”隧道，可以直接通讯。接下来的一行为一个整数k，表示将遭受攻击的星球的
数目。接下来的k行，每行有一个整数，按照顺序列出了帝国军的攻击目标。这k个数互不相同，且都在0到n-1的范
围内。
Output

　　输出文件的第一行是开始时星球的连通块个数。接下来的N行，每行一个整数，表示经过该次打击后现存星球
的连通块个数。
Sample Input
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
Sample Output
1

1

1

2

3

3

解题思路：逆向思维好题。
正着做很麻烦，我们倒着考虑这个问题。首先求出 m 次删除以后的连通块的个数。然后对删除操作进行倒
序处理，如果当前删的这个点不是第一次删除，那么不考虑，否则联通块加一，然后对与这个点相邻的点进行考
虑，有可能还涉及到连通块的减一，用并查集维护就可以了。

代码如下：

#include <bits/stdc++.h>
using namespace std;
const int N = 400010;
int n, m, block;
struct edge{
    int v, nxt;
    edge(){}
    edge(int v, int nxt) : v(v), nxt(nxt) {}
}E[N*2];
int head[N], cnt;
bool vis[N]; //标记在已经存在的图，往里加block减一
bool des[N];
int fa[N];
int q[N];
int ans[N];
int find_set(int x){
    if(x == fa[x]) return x;
    else return fa[x] = find_set(fa[x]);
}
void init(){
    memset(head, -1, sizeof(head));
    cnt = 0;
}
void addedge(int u, int v){
    E[cnt].v = v, E[cnt].nxt = head[u], head[u] = cnt++;
}
void add(int x){
    int p = find_set(x);
    for(int i = head[x]; ~i; i = E[i].nxt){
        int v = E[i].v;
        if(vis[v]){
            int q = find_set(v);
            if(p != q){
                fa[q] = p;
                block--;
            }
        }
    }
}
int main(){
    scanf("%d%d", &n, &m);
    init();
    for(int i = 1; i <= n; i++) fa[i] = i;
    for(int i = 1; i <= m; i++){
        int x, y;
        scanf("%d%d", &x, &y);
        addedge(x, y);
        addedge(y, x);
    }
    scanf("%d", &m);
    for(int i = 1; i <= m; i++){
        scanf("%d", &q[i]);
        des[q[i]] = 1;
    }
    for(int i = 0; i < n; i++){
        if(!des[i]){
            block++;
            add(i);
            vis[i] = 1;
        }
    }
    ans[m + 1] = block;
    for(int i = m; i > 0; i--){
        block++;
        add(q[i]);
        vis[q[i]] = 1;
        ans[i] = block;
    }
    for(int i = 1; i <= m + 1; i++){
        printf("%d\n", ans[i]);
    }
    return 0;
}