L2-013 红色警报（并查集+倒序加点）

题目链接：题目详情 - L2-013 红色警报 (pintia.cn)

样例输入： 

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.

分析：看到这道题的数据范围，n是不大于500的，m是不大于5000的，那么我们每次枚举删除前i个点，那么也就是每次把除了前i个待删除的点之外的所有点加入图中，然后判断他们的连通性作出分析即可。这样的复杂度就是O(k*m)。因为本题对复杂度要求不高，所以这样是可以ac的。但是本着精益求精的态度，我们可以通过一种其他的思路把这道题的复杂度降低至O(m)，方法就是我们倒序加点，因为并查集原则上只能实现合并节点，并不能删除节点，所以我们可以向图中反序加入待删除的点。我们先存下待删除的k个点，把其余没被删除的k个点及其之间的边都存入图中，然后我们先添加最后删除的那个点及与其相关的边（前提是边的两个节点均在图中），添加完这个点后就相当于删除了前k-1个点，然后再添加上倒数第二个删除的点，添加完这个点后就相当于删除了前k-2个点，依次类推，我们就可以通过一次加边从而枚举出删除任意多个点之后的图，总的复杂度就是O(m)，实现上会稍微复杂一点，但是复杂度却被大大降低。

细节见代码：

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<map>
#include<queue>
#include<vector>
#include<cmath>
using namespace std;
const int N=1e5+10;
vector<int> p[N];
int fu[N];
bool vis[N];//vis[i]=true代表第i个城市是存在的 
int find(int x)
{
	if(x!=fu[x]) return fu[x]=find(fu[x]);
	return x;
}
int a[N],b[N];
int ans[N];//ans[i]=1/0表示删除第i个城市会/不会改变其他城市的连通性 
int main()
{
	int n,m;
	cin>>n>>m;
	for(int i=1;i<=n;i++)
		fu[i]=i;
	for(int i=1;i<=m;i++)
	{
		int u,v;
		scanf("%d%d",&u,&v);
		u++,v++;
		p[u].push_back(v);
		p[v].push_back(u);
	}
	memset(vis,true,sizeof vis);
	int k;
	cin>>k;
	for(int i=1;i<=k;i++)
	{
		scanf("%d",&a[i]);
		a[i]++;
		vis[a[i]]=false;
	}
	int cnt=0;//记录连通块个数
	for(int i=1;i<=n;i++)//先把过程中未被破坏的边加入图中 
	{
		if(!vis[i]) continue;
		cnt++;
		for(int j=0;j<p[i].size();j++)
		{
			if(vis[p[i][j]])
			{
				int f1=find(i),f2=find(p[i][j]);
				if(f1==f2) continue;
				fu[f1]=f2;
				cnt--;
			}
		}
	}
	int tcnt=cnt;
	for(int i=k;i>=1;i--)
	{
		int u=a[i];
		cnt++;
		for(int j=0;j<p[u].size();j++)
		{
			if(vis[p[u][j]])
			{
				int f1=find(u),f2=find(p[u][j]);
				if(f1==f2) continue;
				fu[f1]=f2;
				cnt--;
			}
		}
		if(tcnt>cnt) ans[i]=1;
		vis[u]=true;
		tcnt=cnt;
	}
	for(int i=1;i<=k;i++)
	{
		if(ans[i]) printf("Red Alert: City %d is lost!\n",a[i]-1);
		else printf("City %d is lost.\n",a[i]-1);
	}
	if(k==n) printf("Game Over.");
	return 0;
}