[BZOJ1015][JSOI2008]星球大战starwar

[JSOI2008]星球大战starwar

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

Solution
离线乱搞,时间倒流

Code

#include <bits/stdc++.h>
using namespace std;
#define rep(i, l, r) for (int i = l; i <= r; i++)
#define per(i, r, l) for (int i = r; i >= l; i--)
#define MS(_) memset(_, 0, sizeof(_))
#define MP make_pair
#define PB push_back
#define debug(...) fprintf(stderr, __VA_ARGS__)
typedef long long ll;
typedef pair<int, int> PII;
template<typename T> inline void read(T &x){
    x = 0; T f = 1; char ch = getchar();
    while (!isdigit(ch)) {if (ch == '-') f = -1; ch = getchar();}
    while (isdigit(ch))  {x = x * 10 + ch - '0'; ch = getchar();}
    x *= f;
}

const int N = 400000 + 100;
const int M = 200000 + 100;
int n, m, k, cnt;
int ans[N], target[N], choose[N], fa[N];
struct Node{ int v; Node *nxt; }pool[M<<2], *tail=pool, *g[N], *h[N];
inline int find(int k) {return fa[k] = fa[k]==k ? k : find(fa[k]);}
inline void merge(int u, int v){fa[find(u)] = find(v);}
inline void addedge(int u, int v){ u++; v++;
    tail->v = v; tail->nxt = g[u]; g[u] = tail++;
    tail->v = u; tail->nxt = g[v]; g[v] = tail++;
}

int main(){ MS(g); MS(choose);
    read(n); read(m);
    rep(i, 1, m){int u, v; read(u); read(v); addedge(u,v);}

    read(k); 
    rep(i, 1, k) read(target[i]), target[i]++;
    rep(i, 1, k) choose[target[i]] = 1;

    cnt = n-k;
    rep(i, 1, n) fa[i] = i;
    rep(i, 1, n) if (!choose[i])
        for (Node *p = g[i]; p; p = p->nxt) 
            if (!choose[p->v] && find(i) != find(p->v)) cnt--, merge(i, p->v);        

    per(i, k, 1){
        ans[i] = cnt++;
        for (Node *p = g[target[i]]; p; p = p->nxt)
            if (!choose[p->v] && find(target[i]) != find(p->v)) cnt--, merge(target[i], p->v);
        choose[target[i]] = 0;
    }

    printf("%d\n", cnt);
    rep(i, 1, k) printf("%d\n", ans[i]);
    return 0;           
}

你可能感兴趣的:(并查集)