L2-013 紅色警報 （25 分）（並查集）

題目連結：

【解題思路】

每一次都要重新判斷一下需要連線的道路變化有無影響

有影響的情況比較難找，那就直接先判斷沒有影響的情況

就是等於前集合數目等於現集合數目或原集合數目+1等於現集合數目（因為題意，若該國本來就不完全連通，是分裂的k個區域，而失去一個城市並不改變其他城市之間的連通性，則不要發出警報）

這樣即使去掉當前結點也無礙

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
#define ll long long
#define maxn 5020
int father[maxn];
int vis[maxn];
struct node {
    int x,y;
}Node[maxn];
void init()
{
    for(int i=0;i<maxn;i++)
    {
        father[i]=i;
    }
}
/*int findFather(int x)
{
    while (x!=father[x])
    {
        x=father[x];
    }
    return x;
}*/
int findFather(int x)
{
    return x==father[x]?father[x]:findFather(father[x]);
}
void Union(int a,int b)
{
    int faA=findFather(a);
    int faB=findFather(b);
    if(faA!=faB)
        father[faA]=faB;
}
int main()
{
    int n,m;
    while (~scanf("%d%d",&n,&m))
    {
        init();
        memset(vis, 0, sizeof(vis));
        for(int i=0;i<m;i++)
        {
            scanf("%d%d",&Node[i].x,&Node[i].y);
            Union(Node[i].x, Node[i].y);
        }
        int num=0;
        for(int i=0;i<n;i++)
        {
            if(father[i]==i)
                num++;
        }
        int k;
        scanf("%d",&k);
        int num1=0;
        while (k--)
        {
            for(int i=0;i<n;i++)
            {
                father[i]=i;//之前攻佔之後的連通情況發生改變，要進行初始化
            }
            num1=0;
            int h;
            scanf("%d",&h);
            vis[h]=1;
            for(int i=0;i<m;i++)
            {
                if(vis[Node[i].x]==0&&vis[Node[i].y]==0)
                    Union(Node[i].x, Node[i].y);
            }
            for(int i=0;i<n;i++)
            {
                if(father[i]==i)
                    num1++;
            }
            if(num1==num||(num+1==num1))//不改變其他城市的連通性
                printf("City %d is lost.\n",h);
            else
                printf("Red Alert: City %d is lost!\n",h);
            num=num1;
        }
        num=0;
        for(int i=0;i<n;i++)
        {
            if(vis[i]==1)
                num++;
        }
        if(num==n)
            printf("Game Over.\n");
    }
    return 0;
}