L2-013 紅色警報 (25分)
戰(zhàn)爭中保持各個城市間的連通性非常重要。本題要求你編寫一個報警程序，當失去一個城市導(dǎo)致國家被分裂為多個無法連通的區(qū)域時，就發(fā)出紅色警報。注意：若該國本來就不完全連通，是分裂的k個區(qū)域，而失去一個城市并不改變其他城市之間的連通性，則不要發(fā)出警報。
輸入格式：

輸入在第一行給出兩個整數(shù)N（0 < N ≤ 500）和M（≤ 5000），分別為城市個數(shù)（于是默認城市從0到N-1編號）和連接兩城市的通路條數(shù)。隨后M行，每行給出一條通路所連接的兩個城市的編號，其間以1個空格分隔。在城市信息之后給出被攻占的信息，即一個正整數(shù)K和隨后的K個被攻占的城市的編號。
注意：輸入保證給出的被攻占的城市編號都是合法的且無重復(fù)，但并不保證給出的通路沒有重復(fù)。
輸出格式：

對每個被攻占的城市，如果它會改變整個國家的連通性，則輸出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> using namespace std; const int maxn1 = 500; const int maxn2 = 5005; int count1 = 0;//統(tǒng)計當前的聯(lián)通分量個數(shù) int pre[maxn1]; struct edge {int x, y; }e[maxn2]; int find(int x)//并查集模板，查找 {while (x != pre[x]) x = pre[x];return x; } void merge(int x, int y)//并查集模板，合并 {int a = find(x);int b = find(y);if (a != b) pre[a] = b; } void init() {for (int i = 0; i < maxn1; i++) pre[i] = i;//初始化pre數(shù)組 } int main() {int n, k, x, y;cin >> n >> k;init();for (int i = 0; i < k; i++)//將邊存入結(jié)構(gòu)體{cin >> x >> y;e[i].x = x;e[i].y = y;merge(x, y);//合并邊}for (int i = 0; i < n; i++){if (pre[i] == i)count1++;//聯(lián)通分類個數(shù)，即獨立的塊數(shù)}int flag[maxn1]{ 0 };int h;cin >> h;for (int i = 0; i < h; i++){cin >> x;//輸入被攻占城市編號init();flag[x] = 1;//標記該城市被攻占int count2 = 0;//記錄被攻占后的城市的聯(lián)通分量個數(shù)for (int z = 0; z < k; z++){if (flag[e[z].x] == 0 && flag[e[z].y] == 0)//如果該道路的兩個城市都未被攻占{merge(e[z].x, e[z].y);//合并}}for (int j = 0; j < n; j++){if (pre[j] == j && flag[j] == 0)//如果未被攻占且pre[j]==j;{count2++;//記錄當前的聯(lián)通分量數(shù)；}}if (count2 > count1)//聯(lián)通分量大于該城市被攻占前的，紅色警報printf("Red Alert: City %d is lost!\n", x);else//小于等于，只是丟失printf("City %d is lost.\n", x);if (count2 == 0)printf("Game Over.\n");count1 = count2;//被攻占后，更新當前的聯(lián)通分量} } 創(chuàng)作挑戰(zhàn)賽新人創(chuàng)作獎勵來咯，堅持創(chuàng)作打卡瓜分現(xiàn)金大獎

總結(jié)

以上是生活随笔為你收集整理的C++学习之路 | PTA（天梯赛）—— L2-013 红色警报 (25分)（带注释）(并查集)（精简)的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。