【JSOI2008】星球大战 (并查集)
题面
Description
很久以前,在一个遥远的星系,一个黑暗的帝国靠着它的超级武器统治者整个星系。某一天,凭着一个偶然的机遇,一支反抗军摧毁了帝国的超级武器,并攻下了星系中几乎所有的星球。这些星球通过特殊的以太隧道互相直接或间接地连接。 但好景不长,很快帝国又重新造出了他的超级武器。凭借这超级武器的力量,帝国开始有计划地摧毁反抗军占领的星球。由于星球的不断被摧毁,两个星球之间的通讯通道也开始不可靠起来。现在,反抗军首领交给你一个任务:给出原来两个星球之间的以太隧道连通情况以及帝国打击的星球顺序,以尽量快的速度求出每一次打击之后反抗军占据的星球的连通快的个数。(如果两个星球可以通过现存的以太通道直接或间接地连通,则这两个星球在同一个连通块中)。
Input
第一行包含两个整数N (2 ≤ N ≤ 2M)和M (1 ≤ M ≤ 200,000),分别表示星球的数目和“以太”隧道的数目。星球用0到N – 1的整数编号。
接下来的M行,每行包含两个整数X和Y (0 ≤ X ≠ Y < N),表示星球X和星球Y之间有“以太”隧道,可以直接通讯。
接下来的一行包含一个整数K,表示将遭受攻击的星球的数目。
接下来的K行,每行一个整数,按照顺序列出了帝国军的攻击目标。这K个数互不相同,且都在0到N – 1的范围内。
Output
第一行是开始时星球的连通块个数。接下来的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
题解
考虑正向来做的话,很显然,无法维护联通快,只能够每一次重新来做一遍并查集(因为你要拆边)
但是,反着来看这道题,每次恢复一个点,这样就可以并查集来搞了。
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
#define MAX 1000000
inline int read()
{
register int x=0,t=1;
register char ch=getchar();
while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
if(ch=='-'){t=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){x=x*10+ch-48;ch=getchar();}
return x*t;
}
struct Line
{
int v,next;
}e[MAX];
int h[MAX],cnt=1,A[MAX];
int f[MAX],tt[MAX];
bool killed[MAX];
int N,M,KK,Ans;
inline void Add(int u,int v)
{
e[cnt]=(Line){v,h[u]};
h[u]=cnt++;
}
int getf(int x)
{
return x==f[x]?x:f[x]=getf(f[x]);
}
int main()
{
N=read();M=read();
for(int i=1;i<=M;++i)
{
int u=read()+1,v=read()+1;
Add(u,v);Add(v,u);
}
KK=read();
for(int i=1;i<=KK;++i)
{
tt[i]=read()+1;
killed[tt[i]]=true;
}
for(int i=1;i<=N;++i)f[i]=i;
for(int i=1;i<=N;++i)
{
if(!killed[i])
{
for(int j=h[i];j;j=e[j].next)
{
if(!killed[e[j].v])
f[getf(i)]=getf(e[j].v);
}
}
}
for(int i=1;i<=N;++i)if(f[i]==i&&!killed[i])++Ans;
A[KK+1]=Ans;
for(int i=KK;i;--i)
{
killed[tt[i]]=false;Ans++;
for(int j=h[tt[i]];j;j=e[j].next)
{
if(killed[e[j].v])continue;
int x=getf(e[j].v),y=getf(tt[i]);
if(x==y)continue;
else
{
f[x]=y;
--Ans;
}
}
A[i]=Ans;
}
for(int i=1;i<=KK+1;++i)
printf("%d\n",A[i]);
return 0;
}