【洛谷P1197】[JSOI2008]星球大战

题目描述

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

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

输入输出格式

输入格式：

输入文件第一行包含两个整数，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]

分析

题目大意：从一个无向图中去掉一些点，求连通块的个数。

知识点：贪心，并查集。

这道题有多个查询，我们可以考虑用离线做法。这就变成了在去掉k个点的基础下，每加一个点连通块的个数。

去掉k个点之后连通块的个数最多是n-k个（每个点互不相通）。我们考虑连通块减少的条件：

如果加入这个点这个点只属于某一个连通块，那么连通块的个数不会减少。

如果加入这个点这个点属于两个连通块，连通块的个数-1，然后再将这两个连通块合并。

这样的话逆序做一遍就行了。

代码

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=400000+5;
inline int read()
{
    int x=0,f=1; char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1; ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0'; ch=getchar();}
    return x*f;
}
int n,m,k,num,tot;
int head[maxn],father[maxn],a[maxn],b[maxn];
bool vis[maxn];
struct node
{
    int from,next,to;
}e[maxn];
inline void add(int from,int to)
{
    e[++num].next=head[from];
    e[num].from=from;
    e[num].to=to;
    head[from]=num;
}
int find(int x)
{
    if(x!=father[x]) father[x]=find(father[x]);
        return father[x];
}
int main()
{
    n=read();m=read();
    for(int i=0;i<n;i++) father[i]=i;
    for(int i=1;i<=m;i++)
    {
        int x,y;
        x=read();y=read();
        add(x,y);add(y,x);
    }
    k=read(); tot=n-k;
    for(int i=1;i<=k;i++) 
    { a[i]=read(); vis[a[i]]=1;}
    for(int i=1;i<=num;i=i+2)//因为建的是双向边
    {
        if(!vis[e[i].from]&&!vis[e[i].to])
        {
            int r1=find(e[i].from);
            int r2=find(e[i].to);
            if(r1!=r2)
            {
                tot--;
                father[r1]=r2;
            }
        }
    }
    b[k+1]=tot;
    for(int j=k;j>=1;j--)
    {
        int u=a[j]; vis[u]=0; tot++;
        int r1=find(u);//这儿是一个小技巧,不更改u的父亲
        for(int i=head[u];i;i=e[i].next)
        if(!vis[e[i].to])
        {
            int r2=find(e[i].to);
            if(r1!=r2){tot--; father[r2]=r1;}
        }//合并的时候把r2合并到r1上，这样u的父亲就不会变，减少计算
        b[j]=tot;
    }
    for(int i=1;i<=k+1;i++)
    printf("%d
",b[i]);
    return 0;
}