L2-013 紅色警報 （25 分）

戰爭中保持各個城市間的連通性非常重要。本題要求你編寫一個報警程式，當失去一個城市導致國家被分裂為多個無法連通的區域時，就發出紅色警報。注意：若該國本來就不完全連通，是分裂的k個區域，而失去一個城市並不改變其他城市之間的連通性，則不要發出警報。

輸入格式：

輸入在第一行給出兩個整數N

注意：輸入保證給出的被攻佔的城市編號都是合法的且無重複，但並不保證給出的通路沒有重複。

輸出格式：

對每個被攻佔的城市，如果它會改變整個國家的連通性，則輸出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<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<cstdio>
#include<queue>
#include<climits>
#include<set>
#include<stack>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
static const int MAX_N = 505;
int n, m;
int pre[MAX_N];		
bool vis[MAX_N];	//判斷是否佔領
pair<int, int> P[5005];	//連通記錄
void init() {
	for (int i = 0; i < n; i++) {
		pre[i] = i;
	}
}
int Find(int x) {
	if (pre[x] != x) return Find(pre[x]);
	return x;
}
void Join(int x, int y) {
	int fx = Find(x), fy = Find(y);
	if (fx != fy) pre[fx] = fy;
}
int main() {
	memset(vis, false, sizeof(vis));
	scanf("%d%d", &n, &m);
	init();
	for (int i = 0; i < m; i++) {
		scanf("%d%d", &P[i].first, &P[i].second);
		Join(P[i].first, P[i].second);
	}
	int cnt = 0;
	for (int i = 0; i < n; i++) {
		if (pre[i] == i) cnt++;
	}
	//for (int i = 0; i < n; i++) printf("%d ", pre[i]);
	int k;
	scanf("%d", &k);
	for (int t = 0; t < k; t++) {
		int u;
		scanf("%d", &u);
		vis[u] = true;
		init();
		for (int i = 0; i < m; i++) {
			int x = P[i].first, y = P[i].second;
			if (!vis[x] && !vis[y]) Join(x, y);	//兩點均未被佔領，則連通(P陣列記錄之前的連通記錄)
		}
		int cur = 0;
		for (int i = 0; i < n; i++) {
			if (!vis[i] && pre[i] == i) cur++;
		}
		printf("%d %d\n", cur, cnt);
		if (cur + 1 == cnt || cur == cnt) printf("City %d is lost.\n", u);	//一個無關點被佔領或一個連通區域內某點被佔領
		else printf("Red Alert: City %d is lost!\n", u);
		cnt = cur;
	}
	if (k == n) printf("Game Over.\n");
	return 0;
}