PAT甲级 1013 Battle Over Cities (25分)

1013 Battle Over Cities (25分)

题目链接：PAT A 1013

思路分析：就是去掉某一个城市以及和它相连的所有边，问添加几条边才能把剩下的城市连接在一起。实际上就是找连通图的问题，如果一个图完全连通，那么需要0条边，如果有两个连通图，那么需要一条边，三个连通图则需要两条边，以此类推~所以本质还是图的遍历问题。在DFS遍历中，如果一个图完全连通，则只需要遍历一次，否则则需要两或多次遍历，所以只需要计算出遍历次数，就可以知道有几个连通图，减去一之后就是需要增加边的个数啦 ~需要注意的是，虽然题目中的意思是删除某一个城市以及和它相连的所有边，但实际上不需要删除，只需要在DFS遍历到这个城市的时候直接return就可以啦，剩下的就是直接套用DFS模板即可。

AC代码：

#include<iostream>
#include<algorithm>
using namespace std;
const int INF = 0x3fffffff; //一个很大的数 
int n, m, k, check;
int g[1000][1000]; //邻接矩阵 
bool vis[1000] = {false}; //记录某个城市是否被访问 
void dfs(int u) {
	if(u == check) //check代表被删除的城市，所以直接返回 
		return;
	vis[u] = true;
	for(int i = 1; i <= n; i++) {
		if(vis[i] == false && g[u][i] != INF)
			dfs(i);
	}
}
int main() {
	fill(g[0], g[0] + 1000 * 1000, INF);
	scanf("%d %d %d", &n, &m, &k);
	int a, b;
	for(int i = 0; i < m; i++) { //建立图 
		scanf("%d %d", &a, &b);
		g[a][b] = 0;
		g[b][a] = 0;
	}
	for(int i = 0; i < k; i++) {
		int num = 0; //每删除一个新的城市要将num重新置零 
		fill(vis, vis + 1000, false); //每删除一个新的城市要将vis数组恢复为初始未访问状态 
		scanf("%d", &check);
		for(int j = 1; j <= n; j++) {
			if(vis[j] == false && j != check) {
				dfs(j);
				num++; //统计连通图个数 
			}
		}
		printf("%d\n", num - 1);
	}
	return 0;
}

图的遍历使用DFS或者BFS均可，大家可以选择个人比较习惯的使用，但是论代码简洁程度还是DFS更方便些。