洛谷 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

分析:并查集,本题吧确实是一个很不错的提,要求求出每一次打击之后反抗军占据的星球的连通快的个数。即不断销毁星球看连通块的存在情况,很容易想到逆向思维(这个题膈应了我大半天,大多数博主都这样说,但我觉得这是种思维,没遇到过,还真不太容易),一步步重建星球。

步骤:1.记录被破坏了的星球,先求出没被破坏的星球的连通块个数。

2.然后在一步步加入星球,再求加入后连通块个数。

3.最后逆向输出即可。

上代码:

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<queue>
#include<set>
using namespace std;
int N, M, k;
const int maxn = 400010;
int fa[maxn];
int head[maxn], X[maxn], Y[maxn];
int flag[maxn], K[maxn];
int e_num;
int ans[maxn];
struct uu
{
	int from;
	int to;
	int next;
}e[2*maxn];
int find(int x)
{

	int r = x;
	while (r != fa[r])
	{
		r = fa[r];
	} 
	int i = x, j;
	while (fa[i] != r)
	{
		j = fa[i];
		fa[i] = r;
		i = j;
	}
	return r;
}
void unite(int x, int y)
{
	
	int fx = find(x);
	int fy = find(y);
	if (fx != fy)
	{
		fa[fx] = fy;
	}
}
void add(int x, int y)
{
	//起计数以及确定值的作用且正反分别记录,便于后期便利查找。
	++e_num;
	e[e_num].from = x;
	e[e_num].to = y;
	e[e_num].next = head[x];
	head[x] = e_num;

}
int main()
{
	scanf("%d%d", &N, &M);
	for (int i = 0; i <= N; i++)
	{
		fa[i] = i;
	}
	for (int i = 0; i < M; i++)
	{
		scanf("%d%d", &X[i], &Y[i]);
		add(X[i], Y[i]);//将两个值在不同方向分别记录。
		add(Y[i], X[i]);
	}
	scanf("%d", &k);
	memset(flag, 0, sizeof(flag));
	for (int i = 0; i < k; i++)
	{
		scanf("%d", &K[i]);
		flag[K[i]] = 1;
	}
	for (int i = 0; i < M; i++)
	{
		if (!flag[X[i]] && !flag[Y[i]])
		{
			unite(X[i], Y[i]);
		}
	}
	int sum = 0;
	for (int i = 0; i < N; i++)
	{
		if (!flag[i] && fa[i] == i)
			sum++;
	}
	int t = 0;
	for (int i = k-1; i >= 0; i--)
	{
		ans[++t] = sum;//将sum存贮在数组中。
		int u = K[i];
		flag[u] = 0;
		sum++;//建立一个星球,联通块加一
		for (int j = head[u]; j; j = e[j].next)
		{
			int v = e[j].to;//因为add两次所以只需to便可
			if (!flag[v])
			{
				if (find(u) != find(v))//父结点不同科合并
				{
					sum--;//合并连通块减一
					unite(u, v);
				}
			}
		}

	}
	printf("%d\n", sum);
	for (int i = t; i > 0; i--)
	{
		printf("%d\n", ans[i]);
	}
	return 0;
}

猜你喜欢

转载自blog.csdn.net/fighting_yifeng/article/details/81389481