PAT L2-013-红色警报（Tarjan算法）

题目

红色警报（25 分）

战争中保持各个城市间的连通性非常重要。本题要求你编写一个报警程序，当失去一个城市导致国家被分裂为多个无法连通的区域时，就发出红色警报。注意：若该国本来就不完全连通，是分裂的k个区域，而失去一个城市并不改变其他城市之间的连通性，则不要发出警报。

输入格式：

输入在第一行给出两个整数N（0 $<$ N $\le$ 500）和M（$\le$ 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.

题目分析

题目给出了一个图（不一定连通）的顶点集合，之后删除其中节点。重点在于判断这个节点是否是该图的割点。这样一来就很容易想到使用“tarjan算法”来求解图中的割点集合，之后判断该点是否存在于割点集合中。最后说一下这个算法的时间复杂度为O(V+E)。

---------------------------------------------------------------------------------------------------------------------------------

下面来简单讲解一下tarjan算法

这个算法建立在DFS之上，在DFS过程中求出“深度优先搜索树（DFS tree）”。以题目为例：

这里是以1为根构造的深度优先搜索树，其中实现我们可以成为父子边，虚线成为回边。

在DFS过程中，如果一个节点 v 可以被节点 u 搜索到，那么我们就把 u 叫做 v 的父节点。在DFS tree中，u 节点若是割点的话一定会满足下面两个条件中的一个。

1. u 节点是根节点（root node），并且至少有2个子节点
2. u 非根节点且非叶子节点（non-root，non-leaf node u），并且他的子节点 v 满足：以 v 为根的子树（subtree）中没有一个节点满足有一条回边能到达 u 的祖先节点。

第一个条件很好处理，这里不在赘述（可参考代码）。这里的难点重点在于第二个条件的处理，那我们要怎么处理它呢？

对于第二个条件，我们可以通过维护一个disc[]数组，其中存储每个结点是第几个被搜索到的序号。之后对于每个结点 u 我们还需要维护一个数组 low[]，其中存储 u 的子树所能到达的最先被搜索到的点。对low[] 的定义如下：

• low[u] = min{ low[u],disc[v] }
• low[u] = min{ low[u],low[v] }

最后，判断一下 low[v]>=disc[u] 为真就说明u点是割点（APs）。

我这里提醒一下，low[] 与 disc[]越小说明越在dfs tree的上方，也就是说越先被搜索到。

----------------------------------------------------------------------------------------------------------------------------------------------------------------------------

那我们现在再回到这道题，要注意的一点就是每次删除点后都要把与该点有关的边也删除。在增添一个数组ap[]来存储割点的编号。

代码如下

#include<iostream> 
using namespace std;

int adj[550][550];//邻接矩阵
int parent[550],disc[550],low[550];
bool visit[550];//是否访问过
bool ap[550];//存储割点
int n;

void dfs(int u){
	static int time = 0;
	int children = 0;
	visit[u] = true;
	disc[u] = low[u] = ++time;
	for(int i=0;i<n;i++){
		if(adj[u][i] != 0){
			int v = i;
			if(!visit[v]){
				children++;
				parent[v] = u;
				dfs(v);
				low[u] = min(low[u],low[v]);
				if(parent[u] == -1 && children>1){ //parent[u] == -1 表示u为父节点
					ap[u] = true;
				}
				if(parent[u] != -1 && low[v] >= disc[u]){
					ap[u] = true;
				}
			}
			else if(parent[u] != v){
				low[u] = min(low[u],disc[v]);
			}
		}	
	}
}
//初始化算法中用到的数组
void init(){
	for(int i=0;i<n;i++){
		parent[i] = -1;
		visit[i] = false;
		ap[i] = false;
		disc[i] = 0;
		low[i] = 0;
	}
}

int main(){
	int m;
	cin>>n>>m;
	for(int i=0;i<m;i++){
		int u,v;
		cin>>u>>v;
		adj[u][v] = 1;
		adj[v][u] = 1;
	}
	int k;
	cin>>k;
	for(int i=0;i<k;i++){
		int node;
		cin>>node;
		init();
		dfs(node);
		if(ap[node]){
			printf("Red Alert: City %d is lost!\n",node);
		}else{
			printf("City %d is lost.\n",node);
		} 
		for(int j=0;j<n;j++){
			adj[node][j] = 0;
			adj[j][node] = 0;
		}
		if(i == n-1){
			printf("Game Over.\n");
		}
	}
	return 0;
}