7-1 红色警报--C++

战争中保持各个城市间的连通性非常重要。本题要求你编写一个报警程序，当失去一个城市导致国家被分裂为多个无法连通的区域时，就发出红色警报。注意：若该国本来就不完全连通，是分裂的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<iostream>
using namespace std;
// 定义全局变量，包括邻接矩阵表示的图 (mep) 和并查集数组 (pre)
int mep[510][510] = { 0 }; // 邻接矩阵，表示城市之间的连接关系
int pre[510]; // 并查集数组，用于管理城市的连通性
int N; // 城市数量
// 初始化并查集，初始时每个城市都指向自己
void aaa(int n) {
    for (int i = 0; i < n; i++) {
        pre[i] = i;
    }
}
// 并查集的查找操作，路径压缩优化
int bbb(int x) {
    if (x == pre[x]) return x;
    else return pre[x] = bbb(pre[x]);
}
// 并查集的合并操作，将两个城市所在的集合合并
void ccc(int x, int y) {
    int xf = bbb(x); // 找到城市 x 的根节点
    int yf = bbb(y); // 找到城市 y 的根节点
    if (xf != yf) pre[xf] = yf; // 如果根节点不同，则合并
}
                   // 检查城市 k 被移除后的连通性变化
int ddd(int k, int count) {
    int count2 = 0; // 记录移除城市后的连通分量数量
    aaa(N);         // 重新初始化并查集
                    // 将城市 k 从邻接矩阵中移除
    for (int i = 0; i < N; i++) {
        mep[i][k] = 0;
        mep[k][i] = 0;
    }
                    // 使用并查集重新计算剩余城市的连通性
    for (int i = 0; i < N; i++) {
        for (int j = i; j < N; j++) {
            if (mep[i][j]) ccc(i, j);
        }
    }
    // 计算连通分量的数量
    for (int i = 0; i < N; i++) {
        if (i == pre[i]) count2++;
    }
    // 根据连通分量的变化输出相应的提示信息
    if (count2 > count + 1)
        cout << "Red Alert: City " << k << " is lost!\n";
    else
        cout << "City " << k << " is lost.\n";
    return count2; // 返回新的连通分量数量
}

int main() {
    int M, K, count = 0, a, b, k;
    int flag = 1;

    cin >> N >> M; // 输入城市数量和道路数量
    aaa(N);        // 初始化并查集
                   // 读取城市之间的道路信息并更新邻接矩阵
    for (int i = 0; i < M; i++) {
        cin >> a >> b;
        mep[a][b] = 1;
        mep[b][a] = 1;
        ccc(a, b); // 更新并查集
    }
                   // 计算初始状态下的连通分量数量
    for (int i = 0; i < N; i++) {
        if (i == bbb(i)) {
            count++;
        }
    }
    cin >> K;             // 输入需要移除的城市数量
    if (K == N) flag = 0; // 如果移除的城市数量等于总数，则设置游戏结束标志
                          // 逐个移除城市并检查连通性变化
    for (int i = 0; i < K; i++) {
        cin >> k;
        count = ddd(k, count);
    }
    if (!flag) cout << "Game Over."; // 如果所有城市都被移除，输出游戏结束
    return 0;
}