L2-013 紅色警報 (25分)

#include<bits/stdc++.h>
using namespace std;
const int maxn=5050;
int n,m,k,a,b,c,cnt1,cnt2,f[maxn],vis[maxn];
struct xy{int x,y;}node[maxn];
int find(int x)	{return x==f[x]?x:f[x]=find(f[x]);}
void merge(int a,int b)	{a=find(a);b=find(b);f[a]=b;}
int main()
{
	cin>>n>>m;
	for(int i=0;i<n;i++)	f[i]=i;
	for(int i=0;i<m;i++)
	{
		cin>>node[i].x>>node[i].y;
		merge(node[i].x,node[i].y);
	}
	for(int i=0;i<n;i++)	find(i);
	for(int i=0;i<n;i++)
		if(f[i]==i)	cnt1++;
	cin>>k;
	while(k--)
	{
		cin>>c;
		vis[c]=1,cnt2=0;
		for(int i=0;i<n;i++)	f[i]=i;
		for(int i=0;i<m;i++)
			if(!vis[node[i].x]&&!vis[node[i].y])
				merge(node[i].x,node[i].y);
		for(int i=0;i<n;i++)	find(i);
		for(int i=0;i<n;i++)
			if(f[i]==i&&!vis[i])	cnt2++;
		if(cnt2==cnt1 || cnt1==cnt2+1) printf("City %d is lost.\n",c);
        else printf("Red Alert: City %d is lost!\n",c);
        cnt1=cnt2;
    }
    cnt1=0;
    for(int i=0;i<n;i++)
    	if(vis[i])	cnt1++;
  if(cnt1==n) printf("Game Over.\n");
}