[JSOI2008]星球大战——并查集+逆向思维

Posted 2020-10-09

题目描述

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

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

输入输出格式

输入格式：

 

输入文件第一行包含两个整数，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
主要思路:
此题可以运用逆向思维，在剩余节点的基础上反向逐个加入被删除的节点，让每个节点跟与它相邻的并且尚未删除的点集中的节点合并，ans数组存每次删除操作后的连通块的个数，最后倒序输出即可。
代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<vector>
using namespace std;
int n,m,k,x,y,z,num,del[400005];
int father[400005];
bool vis[400005];
vector<int>ans;
vector<int>mp[400005];
int read()
{
    char c=getchar();
    while(!isdigit(c)) c=getchar();
    int x=0;
    while(isdigit(c))
    {
        x=x*10+c-‘0‘;
        c=getchar();
    }
    return x;
}
int Find(int x)
{
    //if(father[x]!=x) father[x]=Find(father[x]);
    //return father[x];
    return x==father[x]?x:father[x]=Find(father[x]);
}
int main()
{
    n=read();m=read();
    for(int i=1;i<=m;i++)
    {
        x=read();y=read();
        mp[x].push_back(y);
        mp[y].push_back(x);
    }
    k=read();
    num=n-k;
    for(int i=1;i<=k;i++)
    {
        del[i]=read();
        vis[del[i]]=true;
    }
    for(int i=0;i<n;i++)
    father[i]=i;
    for(int i=0;i<n;i++)
    if(!vis[i])
    {
        int x,y;
        for(int j=0;j<mp[i].size();j++)
        if(!vis[mp[i][j]])
        {
            x=Find(i);
            y=Find(mp[i][j]);
            if(x!=y)
            {
                father[y]=x;
                num--;
            }
        }
    }
    ans.push_back(num);
    for(int i=k;i;i--)
    {
        vis[del[i]]=false;
        num++;
        int x,y;
        for(int j=0;j<mp[del[i]].size();j++)
        {
            if(!vis[mp[del[i]][j]])
            {
                x=Find(del[i]);
                y=Find(mp[del[i]][j]);
                if(x!=y)
                {
                    father[y]=x;
                    num--;
                }
            }
        }
        ans.push_back(num);
    }
    for(int i=ans.size()-1;i>=0;i--)
    cout<<ans[i]<<endl;
    return 0;
}