洛谷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
说明

[JSOI2008]

---

由于题目没有要求在线, 考虑离线做法.
使用并査集. 逆向维护加点.
代码如下. 目前是WA的, 但仍然可以用作参考. 欢迎查错(最主要是这题好难调试的说). A了以后会回来改.

#include<iostream>
#include<string.h>
#include<stdio.h>
using namespace std;
const int maxN = (int)1e5;
const int maxM = (int)2e5;
const int maxK = maxN;
int n;
int head[maxN];
int size;
struct edge
{
    int v, next;
}G[maxM << 1];
void addEdge(int u, int v)
{
    G[size].v = v;
    G[size].next = head[u];
    head[u] = size ++;
}
int Q[maxK];
int tag[maxN];
int pre[maxN];
void getPre(int u)
{
    if(u == pre[u])
        return;
    getPre(pre[u]);
    pre[u] = pre[pre[u]];
}
int rec[maxN];
int ans[maxK + 1];
void count(int x)
{
    memset(rec, 0, sizeof(rec));
    int cnt = 0;
    for(int i = 0; i < n; i ++)
        if(! tag[i])
        {
            getPre(i);
            if(! rec[pre[i]])
                cnt ++;
            rec[pre[i]] ++;
        }
    ans[x] = cnt;
}
int main()
{
    int m;
    cin >> n >> m;
    size = 0;
    memset(head, - 1, sizeof(head));
    for(int i = 0; i < m; i ++)
    {
        int u, v;
        cin >> u >> v;
        addEdge(u, v);
        addEdge(v, u); 
    }
    int K;
    cin >> K;
    memset(tag, 0, sizeof(tag));
    for(int i = 0; i < K; i ++)
        cin >> Q[i], tag[Q[i]] = 1;
    for(int i = 0; i < n; i ++)
        pre[i] = i;
    for(int i = 0; i < n; i ++)
        if(! tag[i])
            for(int j = head[i]; j != -1; j = G[j].next)
            {
                int v = G[j].v;
                if(tag[v])
                    continue;
                getPre(i);
                getPre(v);
                pre[pre[v]] = pre[i];
            }
    count(K);
    for(int i = K - 1; i >= 0; i --)
    {
        int u = Q[i];
        for(int j = head[u]; j != - 1; j = G[j].next)
        {
            int v = G[j].v;
            if(tag[v])
                continue;
            getPre(u);
            getPre(v);
            pre[pre[v]] = pre[u];
        }
        count(i);
        tag[u] = 0;
    }
    for(int i = 0; i <= K; i ++)
        printf("%d
", ans[i]);
}