BZOJ - 1015【并查集】

1015: [JSOI2008]星球大战starwar

Time Limit: 3 Sec  Memory Limit: 162 MB
Submit: 5721  Solved: 2639
[Submit][Status][Discuss]

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

 

题解：逆向并查集即可，学习了新姿势，不用每次去getfa求联通块数目，只需要每次新加点的时候 num++，然后合并的时候num--即可；

代码：

#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdio>
#include <bitset>
#include <vector>
#include <queue>
#include <stack>
#include <cmath>
#include <list>
#include <set>
#include <map>
#define rep(i,a,b) for(int i = a;i <= b;++ i)
#define per(i,a,b) for(int i = a;i >= b;-- i)
#define mem(a,b) memset((a),(b),sizeof((a)))
#define FIN freopen("in.txt","r",stdin)
#define FOUT freopen("out.txt","w",stdout)
#define IO ios_base::sync_with_stdio(0),cin.tie(0)
#define mid ((l+r)>>1)
#define ls (id<<1)
#define rs ((id<<1)|1)
#define N 400010
#define INF 0x3f3f3f3f
#define INFF ((1LL<<62)-1)
using namespace std;
typedef long long LL;
typedef pair<int, int> PIR;
const double eps = 1e-8;
 
int n, m, u, v, t, k[N], f[N], ans[N];
vector <int> G[N];
bool vis[N];
int getfa(int x){
    return f[x] == x ? x : f[x] = getfa(f[x]);
}
int main()
{
    //FIN;
    scanf("%d %d", &n, &m);
    rep(i, 1, m){ 
        scanf("%d %d", &u, &v);
        G[u].push_back(v);
        G[v].push_back(u);
    }
    scanf("%d", &t);
    rep(i, 1, t)    { scanf("%d", &k[i]); vis[k[i]] = true; }
    rep(i, 0, n)    f[i] = i;
 
    int res = n-t;
    rep(i, 0, n-1){
        if(vis[i])  continue;
        int x = getfa(i);
        rep(j, 0, (int)G[i].size()-1){
            if(vis[G[i][j]])    continue;
            int y = getfa(G[i][j]);
            if(x != y)  { f[y] = x; res--; }
        }
    }
    /*int res = 0;
    rep(i, 0, n-1)  { if(!vis[i] && getfa(i) == i)  res++; }*/
    per(i, t, 0){
        ans[i] = res++;
        vis[k[i]] = false;
        int x = getfa(k[i]);
        rep(j, 0, (int)G[k[i]].size()-1){
            if(vis[G[k[i]][j]]) continue;
            int y = getfa(G[k[i]][j]);
            if(x != y){ 
                f[y] = x;
                res--;
            }
        }
    }
    rep(i, 0, t)    printf("%d\n", ans[i]);
    return 0;
}