【并查集或BFS】之L2-013 红色警报 详解

 

并查集思路：

• 首先利用并查集计算出初始时连通块的数量。
• 然后攻掉一个城市，即去掉一个点，再次计算连通块的数量，注意：被去掉的点此刻算作一个连通块，连通块的数量是一直递增的！
• 找到 去掉一个点时连通块数量变化的规律来设置判断是否要发出红色警报。如果被攻占的城市之前是单独一个城市或者是连通块的不影响其他城市连通的城市，cnt=cnt0;如果是连通块的桥的时候，被攻占了cnt=cnt0+2;

关于并查集，这里还有详细的介绍和例子！！ 

https://blog.csdn.net/weixin_43535668/article/details/104413573

https://blog.csdn.net/weixin_43535668/article/details/104436391

#include<iostream>
#include<algorithm>
#include<string>
#include<stdio.h>
using namespace std;
#define N 505
int fa[N];//并查集记录数组
int edge[N][N];
int vis[N];
int lost[N];
void init()
{
	for (int i = 0; i <=N ; i++)
	{
		fa[i] = i;
	}
}
int find(int x)
{
	if (fa[x] == x)
		return x;
	else
		return find(fa[x]);
}
void merge(int x, int y)
{
	int rx = find(x);
	int ry = find(y);
	if (rx != ry)
	{
		fa[rx] = ry;
	}
	
}
int count(int n)//返回连通块的数量
{
	int res = 0;
	for (int i = 0; i < n; i++)
	{
		if (fa[i] == i)
			res++;
	}
	return res;
}

int main()
{
	int n, m, x, y;
	cin >> n >> m;//城市数目 道路数目
	int con[5000][2];
	init();//并查集数组初始化
	for (int i = 0; i < m; i++)
	{
		cin >> con[i][0] >> con[i][1];//记录道路信息
		edge[con[i][0]][con[i][1]] = 1;
		edge[con[i][1]][con[i][0]] = 1;
		merge(con[i][0], con[i][1]);
	}
	int t, cnt0 = count(n);//返回连通块的数量
	cin >> t;
	while (t--)
	{
		cin >> x;
		lost[x] = 1;
		init();
		for (int i = 0; i < m; i++)
		{
			if (!(lost[con[i][0]] || lost[con[i][1]]))//重新统计丢失城市后的连通块
			{
				merge(con[i][0], con[i][1]);
			}
		}
		for (int i = 0; i < m; i++) {
			if (lost[con[i][0]] || lost[con[i][1]]);
			else merge(con[i][0], con[i][1]);
		}
		int cnt = count(n);
		if (cnt == cnt0 + 1 || cnt <= cnt0)//这个判断条件要注意
//如果被攻占的城市之前是单独一个城市或者是连通块的不影响其他城市连通的城市，cnt=cnt0;
//如果是连通块的桥的时候，被攻占了cnt=cnt0+2;
		{
			printf("City %d is lost.\n", x);
		}
		else printf("Red Alert: City %d is lost!\n", x);
		cnt0 = cnt;
	}
	if(cnt0==n)
	 printf("Game Over.\n");
	
	return 0;
}

BFS思路：

使用BFS统计联通分量时，没有计算失守的城市，若本次连通分量个数大于上一次的则要红色警报，否则只需要打印lost即可。

思路和上面大同小异，就是计算连通块数量。

#include<iostream>
#include<queue>
#include<algorithm>
#include<stdio.h>
#include<string>
#include<string.h>
using namespace std;
#define N 505
int edge[N][N], vis[N], lost[N], n;
queue<int> qu;
void bfs()
{
	while (!qu.empty())
	{
		int e = qu.front();
		qu.pop();
		vis[e] = 1;
		for (int i = 0; i < n; i++)
		{
			if ((edge[e][i] || edge[i][e]) && !vis[i] && !lost[i])
			{
				qu.push(i);
				vis[i] = 1;
			}
		}
	}
}
int main()
{
	int m, x, y, t, precnt = 0, cnt = 0;
	cin >> n >> m;//城市数量 道路数量
	for (int i = 0; i < m; i++)
	{
		cin >> x >> y;
		edge[x][y] = 1; edge[y][x] = 1;
	}
	for (int i = 0; i < n; i++)
	{
		if (!vis[i])
		{
			vis[i] = 1;
			qu.push(i);//进队列
			bfs();
			precnt++;//计算连通块的个数
		}
	}
	cin >> t;
	while (t--)
	{
		memset(vis, 0, sizeof(vis));
		cnt = 0;
		cin >> x;
		lost[x] = 1;
		for (int i = 0; i < n; i++)
		{
			if (!vis[i] && !lost[i])
			{
				vis[i] = 1;
				qu.push(i);
				bfs();
				cnt++;
			}
		}
		if(cnt<=precnt)//这里的判断条件与并查集的不同，这里的城市被攻占了就是去掉了
			printf("City %d is lost.\n", x);
		else printf("Red Alert: City %d is lost!\n", x);
		precnt = cnt;
	}
	if(precnt==0)
		printf("Game Over.\n");
	system("pause");
	return 0;
}