【BZOJ1015】【JSOI2008】星球大战starwar

题目描述

很久以前，在一个遥远的星系，一个黑暗的帝国靠着它的超级武器统治者整个星系。某一天，凭着一个偶然的
机遇，一支反抗军摧毁了帝国的超级武器，并攻下了星系中几乎所有的星球。这些星球通过特殊的以太隧道互相直
接或间接地连接。 但好景不长，很快帝国又重新造出了他的超级武器。凭借这超级武器的力量，帝国开始有计划
地摧毁反抗军占领的星球。由于星球的不断被摧毁，两个星球之间的通讯通道也开始不可靠起来。现在，反抗军首
领交给你一个任务：给出原来两个星球之间的以太隧道连通情况以及帝国打击的星球顺序，以尽量快的速度求出每
一次打击之后反抗军占据的星球的连通快的个数。（如果两个星球可以通过现存的以太通道直接或间接地连通，则
这两个星球在同一个连通块中）。

输入

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

输出

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

题解

大概思路就是，先求出全部删点以后还有多少个联通分量，然后一步步往上增点，这样就能由“省选”题转化成“并查集水题”了。当然还有一个难题，就是联通分量个数的查询。一开始我是用的直接枚举，结果Time Limit Exceeded了。后来一想，合并的时候是把两个连通分量合并成一个直接加一个cnt计数，每次合并操作后－1cnt初值为n-k，每增一个点联通分量要+1（因为被删掉的点不在任何一个连通分量里），然后再增边。

My Code

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<vector>
using namespace std;
int n,m,k,fa[400001],cuihui[400001],cnt,pcnt=0,chsx[400001],ans[400001];
vector<int> mp[400001];
int find(int x) {
 if(fa[x]==x)return x;
 fa[x]=find(fa[x]);
 return fa[x];
}
void pushedge(int x1,int y1) {
 int xx=find(x1);
 int yy=find(y1);
 if(xx!=yy) {
  fa[xx]=yy;
  --cnt;
 }
}
int main() {
 memset(cuihui,0,sizeof(cuihui));
 scanf("%d%d",&n,&m);
 int x,y;
 for(int i=1; i<=m; i++) {
  scanf("%d%d",&x,&y);
  x+=1;
  y+=1;
  mp[x].push_back(y);
  mp[y].push_back(x);
 }
 for(int i=1; i<=n; i++) {
  fa[i]=i;
 }
 scanf("%d",&k);
 for(int i=1; i<=k; i++) {
  scanf("%d",&x);
  x+=1;
  chsx[i]=x;
  cuihui[x]=1;
 }
 cnt=n-k;
 for(int i=1; i<=n; i++) {
  if(!cuihui[i]) {
   for(int j=0;j<mp[i].size(); j++) {
    if(!cuihui[mp[i][j]]) {
     pushedge(i,mp[i][j]);
    }
   }
  }
 }
 ans[++pcnt]=cnt;
 for(int i=1; i<=k; i++) {
  cnt++;
  cuihui[chsx[k-i+1]]=0;
  for(int j=0; j<mp[chsx[k-i+1]].size(); j++) {
   if(!cuihui[mp[chsx[k-i+1]][j]]) {
    pushedge(chsx[k-i+1],mp[chsx[k-i+1]][j]);
   }
  }
  ans[++pcnt]=cnt;
 }
 for(int i=pcnt;i>=1;i--){
  printf("%d\n",ans[i]);
 }
}