PTA團體程序設計天梯賽-練習集 L2-013 紅色警報（圖的連通分量）

戰爭中保持各個城市間的連通性非常重要。本題要求你編寫一個報警程序，當失去一個城市導致國家被分裂爲多個無法連通的區域時，就發出紅色警報。注意：若該國本來就不完全連通，是分裂的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.
思路：用dfs統計圖的連通分量有幾個，如果這個城市被攻佔後的連通分量大於被攻佔前的連通分量加一（被攻佔的城市本身是一個連通分量），則這個城市會改變整個國家的連通性輸出Red Alert: City k is lost!，否則輸出City k is lost.，做時要注意佔領前連通分量cnt的更新。

#include <bits/stdc++.h>
using namespace std;
int a[505][505],n,m,t,vis[505];
void dfs(int k){
	for(int i = 0;i < n;i++){
		if(a[k][i] == 1 && vis[i] == 0){
			vis[i] = 1;
			dfs(i);
		}
	}
	return ;
}
int count(){
	int res = 0;
	memset(vis,0,sizeof(vis));
	for(int i = 0;i < n;i++){
		if(vis[i] == 0){
			vis[i] = 1;
			dfs(i);
			res++;
		}
	}
	return res;
}
int main(){
	cin>>n>>m;
	for(int i = 0;i < m;i++){
		int x,y;
		cin>>x>>y;
		a[x][y] = 1;
		a[y][x] = 1;
	}
	int cnt = count();
	cin>>t;
	for(int i = 0;i < t;i++){
		int x;
		cin>>x;
		for(int j = 0;j < n;j++){
			if(a[x][j] == 1){
				a[x][j] = 0;
				a[j][x] = 0;
			}
		}
		int cnt2 = count();
		if(cnt2 > cnt+1){
			printf("Red Alert: City %d is lost!\n",x);
		}else{
			printf("City %d is lost.\n",x);
		}
		//更新cnt
		cnt = cnt2;
		if(i == n-1) printf("Game Over.\n");
	}
	return 0;
}