PTA:7-123 红色警报 (25分)(并查集+解析)

7-123 红色警报 (25分)

战争中保持各个城市间的连通性非常重要。本题要求你编写一个报警程序,当失去一个城市导致国家被分裂为多个无法连通的区域时,就发出红色警报。注意:若该国本来就不完全连通,是分裂的k个区域,而失去一个城市并不改变其他城市之间的连通性,则不要发出警报。

输入格式:
输入在第一行给出两个整数N(0 < N ≤ 500)和M(≤ 5000),分别为城市个数(于是默认城市从0到N-1编号)和连接两城市的通路条数。随后M行,每行给出一条通路所连接的两个城市的编号,其间以1个空格分隔。在城市信息之后给出被攻占的信息,即一个正整数K和随后的K个被攻占的城市的编号。

注意:输入保证给出的被攻占的城市编号都是合法的且无重复,但并不保证给出的通路没有重复。

输出格式:
对每个被攻占的城市,如果它会改变整个国家的连通性,则输出Red Alert: City k is lost!,其中k是该城市的编号;否则只输出City k is lost.即可。如果该国失去了最后一个城市,则增加一行输出Game Over.。

输入样例:
5 4
0 1
1 3
3 0
0 4
5
1 2 0 4 3

输出样例:
City 1 is lost.
City 2 is lost.
Red Alert: City 0 is lost!
City 4 is lost.
City 3 is lost.
Game Over.

思路:
首先利用并查集计算出初始时连通块的数量。
然后攻掉一个城市,即去掉一个点,再次计算连通块的数量。
找到 去掉一个点时连通块数量变化的规律来设置判断是否要发出红色警报。

#include<bits/stdc++.h>
using namespace std;
int n,m,mp[520][520],p[520];
/*
主要思路就是利用并查集。
这里提一下:为什么判断条件是cnt2>cnt+1,输出警报信息
即当去掉i点后,多增加两个连通块

第一个:去掉i点后,一个连通块被分为两个,增加了一个
第二个:去掉i点后,再次统计cnt2时,就会把去掉的点i也加上;
即增加了两个连通块时,需警报 

*/
void init(){
	for(int i=0; i<n; i++)
		p[i]=i;
} 
 
int find(int x){
	if(x==p[x]) return x;
	return p[x]=find(p[x]);
} 

void merge(int x, int y){
	int xx=find(x);
	int yy=find(y);
	if(xx!=yy) p[xx]=yy;
	return;
}

 
int main(){
	int a,b,k,x,cnt=0,cnt2,flag=0;
	cin>>n>>m;
	init();
	while(m--){
		cin>>a>>b;
		mp[a][b]=1;
		mp[b][a]=1;
		merge(a,b);
	}
	for(int i=0; i<n; i++){  //统计连通块个数 
		if(i==find(i)) cnt++;
	}
	cin>>k;
	if(k==n) flag=1; //要输出Game Over; 
	while(k--){
		cin>>x;
		init();
		cnt2=0;  //x去掉后的连通区域个数 
		for(int i=0; i<n; i++) mp[x][i]=mp[i][x]=0;
		for(int i=0; i<n; i++){
			for(int j=0; j<n; j++){
				if(mp[i][j]) merge(i,j);
			}
		}
		for(int i=0; i<n; i++){  //统计连通块个数 
			if(i==find(i)) cnt2++;
		}
		if(cnt2<=cnt+1) cout<<"City "<<x<<" is lost.\n";
		else cout<<"Red Alert: City "<<x<<" is lost!\n";		
		cnt=cnt2;
	}
	if(flag) cout<<"Game Over."<<endl;
	return 0;	
}

欢迎大家批评改正!!!

发布了49 篇原创文章 · 获赞 2 · 访问量 3907

猜你喜欢

转载自blog.csdn.net/weixin_43581819/article/details/104107139