團體程序設計天梯賽-練習集 L2-013. 紅色警報 解題報告

L2-013. 紅色警報

時間限制

400 ms

內存限制

65536 kB

代碼長度限制

8000 B

判題程序

Standard

作者

陳越

戰爭中保持各個城市間的連通性非常重要。本題要求你編寫一個報警程序，當失去一個城市導致國家被分裂爲多個無法連通的區域時，就發出紅色警報。注意：若該國本來就不完全連通，是分裂的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 <cstdio>
#include <cstring>
#include <iostream>
#include <set>
#include <algorithm>
#include <queue>
using namespace std;

int f[550], t, k, n, m, a[5500], b[5500], vis[550];

int find(int x)
{
    return x == f[x] ? x : f[x] = find(f[x]);
}

void merge(int x,int y)
{
    x = find(x);
    y = find(y);
    if(x != y) f[x] = y;
}

int main()
{
    int sum = 0, ts;
    for(int i = 0; i < 550; i++) f[i] = i;
    scanf("%d%d",&n, &m);
    for(int i = 0; i < m; i++)
    {
        scanf("%d%d", &a[i], &b[i]);
        merge(a[i], b[i]);
    }
    for(int i = 0; i < n; i++)
        if(f[i] == i) sum++;
    scanf("%d", &k);
    for(int i = 0; i < k; i++)
    {
        scanf("%d", &t);
        vis[t] = 1;
        ts = 0;
        for(int j = 0; j < n; j++)
            f[j] = j;
        for(int j = 0; j < m; j++)
        {
            if(!vis[a[j]] && !vis[b[j]]) merge(a[j], b[j]);
        }
        for(int j = 0; j < n; j++)
            if(f[j] == j && !vis[j]) ts ++;
        if(ts > sum) printf("Red Alert: City %d is lost!\n", t);
        else printf("City %d is lost.\n", t);
        sum = ts;

    }
    sum = 0;
    for(int i = 0; i < n; i++)
        if(vis[i]) sum++;
    if(sum == n) printf("Game Over.\n");
    return 0;
}