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.

做法:

每删除一个点,计算一次连通块的个数,注意,已失去的点,就不再考虑了

代码:

#include 
using namespace std;
typedef long long LL;
const int N = 510,M = 10010;
int n, m, q;
int h[N], e[M], ne[M], idx;
bool st[N], lost[N];
int cnt, sum;
void add(int a, int b) {
    e[idx] = b, ne[idx] = h[a], h[a] = idx++;
}
void dfs(int u) {
    st[u] = true;
    for (int i = h[u]; ~i; i = ne[i]) {
        int j = e[i];
        if (lost[j])continue;
        if (!st[j])dfs(j);
    }
}
int Count() {
    int res = 0;
    memset(st, 0, sizeof st);
    for (int i = 0; i < n; i++) {
        if (lost[i])continue;
        else {
            if (!st[i]) {
                dfs(i);
                res++;
            }
        }
    }
    return res;
}


int main() {
    scanf("%d%d", &n, &m);
    int a, b, x;
    memset(h, -1, sizeof h);
    for (int i = 1; i <= m; i++) {
        scanf("%d%d", &a, &b);
        add(a, b);
        add(b, a);
    }
    scanf("%d", &q);
    memset(lost, 0, sizeof lost);
    sum = Count();

    for (int i = 0; i < q; i++) {
        scanf("%d", &x);
        lost[x] = true;
        cnt = Count();
        if (cnt > sum) {
            cout << "Red Alert: City " << x << " is lost!" << endl;
        }
        else cout << "City " << x << " is lost." << endl;
        sum = cnt;
    }
    if (q == n)cout << "Game Over.";
}

你可能感兴趣的:(图论)