数据结构:并查集

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

输入
输入文件第一行包含两个整数,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行,每行一个整数,表示经过该次打击后现存星球
的连通块个数。

样例输入
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
2
3
3

#include<iostream>
using namespace std;
const int MAX = 40004;
int f[MAX], a[MAX], head[MAX], num = 0 ,rest[MAX] ;//num:连通统计 f:并查集 a:存被摧毁的点 
bool dis[MAX];
struct edge {
	int from;
	int to;
	int next;
}save[MAX];//存联通边
void Union(int u,int v)
{
	save[num].from = u;
	save[num].next = head[u];
	save[num].to = v;
	head[u] = num;
	num++;//head 存的是a中第几个是u的下一个联通节点
	//先存head,因为初始两个节点肯定没有其他连通量,所以都存-1,但之后的操作head[u]=num;保留了节点的头的序号。因为有正向有反向,相当于给head初始化。
};
int find(int x)
{
	if (x != f[x]) f[x] = find(f[x]);
	return f[x];
}
int main()
{
	int m, n,k;
	cin >> n >> m;
	int u, v;
	
	for (int i = 0; i < n; i++)
	{
		f[i] = i;
		head[i] = -1;
	}
	for (int i = 0; i < m; i++)
	{
		cin >> u >> v;
		Union(u, v); Union(v, u);
	}
	int sum,x;
	cin >> k;
	sum = n - k;
	for (int i=1; i <= k; i++)
	{
		cin >> x;
		dis[x] = true;
		a[i] = x;
	}
	for (int i = 0; i < 2 * m; i++)
	{
		if (dis[save[i].from] == false && dis[save[i].to] == false)
		{
			if (find(save[i].from) != find(save[i].to))
			{
				sum--;
				f[find(save[i].from)] = f[find(save[i].to)];
			}
		}
	}
	rest[k + 1] = sum;
	for (int t = k; t >= 1; t--)
	{
		u = a[t];
		sum++;
		dis[u] = false;
		for (int i = head[u]; i != -1; i=save[i].next)
		{
			if (dis[save[i].to] == false && f[find(u)] != find(save[i].to))
			{
				sum--;
				f[find(save[i].to)] = f[find(u)];
			}
		}
		rest[t] = sum;
	}
	for (int i = 1; i <=k+1; i++)cout << rest[i] << endl;
}
	int m, n,k;
	cin >> n >> m;
	int u, v;
	
	for (int i = 0; i < n; i++)
	{
		f[i] = i;
		head[i] = -1;
	}
	for (int i = 0; i < m; i++)
	{
		cin >> u >> v;
		Union(u, v); Union(v, u);
	}
	int sum,x;
	cin >> k;
	sum = n - k;
	for (int i=1; i <= k; i++)
	{
		cin >> x;
		dis[x] = true;
		a[i] = x;
	}
	for (int i = 0; i < 2 * m; i++)
	{
		if (dis[save[i].from] == false && dis[save[i].to] == false)
		{
			if (find(save[i].from) != find(save[i].to))
			{
				sum--;
				f[find(save[i].from)] = f[find(save[i].to)];
			}
		}
	}
	rest[k + 1] = sum;
	for (int t = k; t >= 1; t--)
	{
		u = a[t];
		sum++;
		dis[u] = false;
		for (int i = head[u]; i != -1; i=save[i].next)
		{
			if (dis[save[i].to] == false && f[find(u)] != find(save[i].to))
			{
				sum--;
				f[find(save[i].to)] = f[find(u)];
			}
		}
		rest[t] = sum;
	}
	for (int i = 1; i <=k+1; i++)cout << rest[i] << endl;
}

最开始输入是全屏RE,发现是数组有点小,而且输入数据太多,又没用c,直接上get的新技能:std::ios::sync_with_stdio(false);

求解连通块问题还是并查集比较好,算法时间复杂度直逼O(n)。
建立并查集无非就是路径压缩,按秩合并(其实完全可以不进行状态压缩)这道题正向的话需要每次摧毁判断连通分量,所以采用逆向方法直接联通分量。

路径压缩:
void find(int x)
{
if(x!=f[x]) f[x]=find[f(x)];
return f[x]
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值