战争中保持各个城市间的连通性非常重要。本题要求你编写一个报警程序,当失去一个城市导致国家被分裂为多个无法连通的区域时,就发出红色警报。注意:若该国本来就不完全连通,是分裂的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.
并查集问题,一开始完全没思路,看了看网上的代码吓坏了,这么长,花了较长时间分析了一下,其实就是找连通集的问题,如果此城市被攻克,整个连通集个数不变,说明只是单单城市被攻破,没什么影响,如果连通集的个数增加了两个及以上,说明此时被分割成了多个部分,需要输出红色警报。
查并集的思路就是遍历每个father数组,如果是根节点,就说明有一个连通集,记录下来,每次被攻克完需要重新更新,由于每次都需要更新,所以需要提前将各个城市的联通路径都存下来,每次城市攻破的时候,再重新Union回去。
如果想要做出此题用并查集的方法的话,需要先去来了解并查集是什么,再做几个简单的例题,如果直接上手此题,难度太大,不易理解
#include<iostream>
#include<string>
#include<algorithm>
#include<bits/stdc++.h>
#include<stack>
#include<set>
#include<vector>
#include<map>
#include<queue>
#include<deque>
#include<cctype>
#include<unordered_set>
#include<unordered_map>
#include<fstream>
using namespace std;
int n,m;
const int maxn = 501;
int father[maxn];//father数组,并查集都需要的
struct Node{//此结构体需要存储各个城市的联通路径
int u,v;
}edge[5001];
int visit[maxn];
int findFather(int x){//找到根节点,我这里用了路径优化,不用的话可能会超时把
if(x==father[x]) return father[x];
else{
int F=findFather(father[x]);//路径优化
father[x]=F;
return F;
// return findFather(father[x]);
}
}
void Union(int a,int b){//合并
int pa = findFather(a);
int pb = findFather(b);
if(pa!=pb){
father[pa]=pb;
}
}
void init(){//记得初始化,每一个节点都是根节点
for(int i=0;i<n;i++){
father[i]=i;
}
}
int main(){
cin>>n>>m;
init();
for(int i=0;i<m;i++){
int a,b;
cin>>a>>b;
edge[i].u=a;//存储联通路径
edge[i].v=b;
Union(a,b);//合并
}
int num1=0,num2=0;
for(int i=0;i<n;i++){
if(i==father[i]){
num1++;//算出原始没攻打情况下,连通集的个数
}
}
int k;
cin>>k;
while(k--){//开始攻克城市
num2=0;
init();//千万别忘了num2和连通集都要初始化
int c;
cin>>c;
visit[c]=1;//攻下来的,标记一下
for(int i=0;i<m;i++){
if(visit[edge[i].u]==1||visit[edge[i].v]==1){//如果此城市被攻破,说明所有的道路都没了
continue;//直接下一次循环即可
}
Union(edge[i].u,edge[i].v);//重新合并
}
for(int i=0;i<n;i++){
if(i==father[i]){//再次计算新的连通集个数
num2++;
}
}
if(num1==num2||num1+1==num2){//如果相等或者只多了一个,没影响
printf("City %d is lost.\n",c);
}else{//如果多出了好几个,说明红色警报了
printf("Red Alert: City %d is lost!\n",c);
}
num1=num2;
}
num1=0;
for(int i=0;i<n;i++){
if(visit[i]==1){//如果全攻克完,游戏结束
num1++;
}
}
if(num1==n) cout<<"Game Over.";
return 0;
}