并查集 洛谷 P1197 [JSOI2008]星球大战

题目描述

很久以前，在一个遥远的星系，一个黑暗的帝国靠着它的超级武器统治者整个星系。

某一天，凭着一个偶然的机遇，一支反抗军摧毁了帝国的超级武器，并攻下了星系中几乎所有的星球。这些星球通过特殊的以太隧道互相直接或间接地连接。

但好景不长，很快帝国又重新造出了他的超级武器。凭借这超级武器的力量，帝国开始有计划地摧毁反抗军占领的星球。由于星球的不断被摧毁，两个星球之间的通讯通道也开始不可靠起来。

现在，反抗军首领交给你一个任务：给出原来两个星球之间的以太隧道连通情况以及帝国打击的星球顺序，以尽量快的速度求出每一次打击之后反抗军占据的星球的连通快的个数。（如果两个星球可以通过现存的以太通道直接或间接地连通，则这两个星球在同一个连通块中）。

输入输出格式

输入格式：

输入文件第一行包含两个整数， $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$ 的范围内。

输出格式：

第一行是开始时星球的连通块个数。接下来的 $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]

算法分析：

倒叙并查集，并查集删除元素是很难且很麻烦的，这题我们可以倒叙的建立并查集，第一步建立图，把每个城市的关系表达在一张图上，然后我们先把不能被炸的城市建立并查集，计算出连通块数量，然后倒着依次加入被炸城市的并查集，用sum数组记录答案。

过程：

1. 读入所有数据，建立图（因为要离线操作）

2. 统计一次删掉所有被毁灭星球后的连通块数量并求并查集

3. 把被炸的星球倒序一个一个加入，求并查集并计算连通块数量

4. 依次输出连通块的数量

代码实现：

#include<bits/stdc++.h>
using namespace std;
const int N=4e6+5;
vector<int>g[N];
int fa[N],a[N],sum[N];
bool vis[N];
int find(int x)
{
    return x==fa[x]?x:fa[x]=find(fa[x]);
}
int main()
{
   int n,m;
   scanf("%d%d",&n,&m);
   for(int i=0;i<=n;i++)
   fa[i]=i;
   for(int i=0;i<m;i++)//先建立图
   {
       int x,y;
       scanf("%d%d",&x,&y);
       x++;y++;
       g[x].push_back(y);
       g[y].push_back(x);
   }
   int k;

   scanf("%d",&k);
   for(int i=1;i<=k;i++)//输出要摧毁的星球
     {
        scanf("%d",&a[i]);
        a[i]++;
        vis[a[i]]=1;    //标记要炸毁的星球
     }
   int ans=0;
   for(int i=1;i<=n;i++) //建立不能被炸毁星球的并查集
   {
       if(vis[i]==0)
       {
           for(int j=0;j<g[i].size();j++)
           {
               if(vis[g[i][j]]==0)
               {
                   int r1=find(i);
                   int r2=find(g[i][j]);
                      fa[r1]=r2;
               }

           }
       }
   }
   //接下来要逆序了
   for(int i=1;i<=n;i++)//累计全被被炸毁星球的连通块数量
   if(fa[i]==i&&vis[i]==0)
   {
       ans++;
   }

   sum[k]=ans;
   for(int i=k;i>=1;i--)//倒着建立被炸毁的星球
   {
     vis[a[i]]=0;   
     ans++;
     for(int j=0;j<g[a[i]].size();j++)
     if(vis[g[a[i]][j]]==0)
     {
       int r1=find(a[i]);
       int r2=find(g[a[i]][j]);
       if(r1!=r2) //如果不在一个连通块
      {
      fa[r1]=r2;
      ans--;
      }
     }
     sum[i-1]=ans;
   }
   for(int i=0;i<=k;i++)
    printf("%d\n",sum[i]);
  return 0;
}