PTA搜尋專題部分題解

7-7 紅色警報 (25分)
戰爭中保持各個城市間的連通性非常重要。本題要求你編寫一個報警程式，當失去一個城市導致國家被分裂為多個無法連通的區域時，就發出紅色警報。注意：若該國本來就不完全連通，是分裂的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.

這道題挺考驗思維的（菜鳥的自我感覺），之前做的都是模板題，雖然專題是“搜尋”，我知道要用bfs/dfs，但是還是沒能做出來，賽後補題的時候仔細想了想才想出來。

解題思路：找出連通分量個數，如果去掉City x會使得連通分量增加至少2以上，就說明要發出紅色警報了。（解題關鍵！！！）

#include <bits/stdc++.h>

using namespace std;

int n,m,k,sum1,sum2;
bool book[510],e[510][510];

void dfs(int u)//dfs找連通量個數
{
    int i;
    for(i=0; i<n; i++)
        if(e[u][i]==1&&book[i]==0)
        {
            book[i]=1;
            dfs(i);
        }
    return;
}

int main()
{
    int i,x,y,p;
    scanf("%d%d",&n,&m);
    for(i=0; i<m; i++)
    {
        scanf("%d%d",&x,&y);
        e[x][y]=e[y][x]=1;
    }
    for(i=0; i<n; i++)//先計算連通量個數
        if(book[i]==0)
        {
            dfs(i);
            sum1++;
        }
    scanf("%d",&k);
    int ans=k;
    while(k--)
    {
        sum2=0;
        scanf("%d",&p);
        for(i=0; i<n; i++)//把p點從“圖”中刪除，注意：這裡不能用book[p]=1，因為後面我們會重新把book賦值0，如果這裡用book[p]=1的話會對答案產生影響
            e[p][i]=e[i][p]=0;
        memset(book,0,sizeof(book));
        for(i=0; i<n; i++)//計算把p從“圖”中刪除後的連通量個數
            if(book[i]==0)
            {
                dfs(i);
                sum2++;
            }
        if(sum2-sum1>=2)
            printf("Red Alert: City %d is lost!\n",p);
        else
            printf("City %d is lost.\n",p);
        sum1=sum2;//把前一次的連通個數賦值給下一次迴圈
    }
    if(ans==n)
        printf("Game Over.\n");
    return 0;
}