PAT 1013 Battle Over Cities (25分)

考点

dfs、连通子图

思路

       题目大意是，当一个城市被占领时，与它相邻的所有高速公路同时被占领，也就是该城市成为图中的一个独立点。给定图中N个城市和M条高速公路，依次判断这K个城市被敌人占领时，最少需要额外修多少条高速公路才能让剩下的城市相连。最少的公路数，即为连通子图的个数-1。因此总的来说，题目考察的就是连通子图的求法。

        最容易想到的一种做法就是，若判断城市c1，则将邻接矩阵中与之相关的均置为1。但是，还有一个更为简单的做法。就是将该点的visit置为true，这样自然就相当于将该点独立出去。

      我很不解的一点在于，当我将scanf换成cin输入数据时，最后一个测试点运行超时。

 

#include <iostream>
#include <algorithm>
using namespace std;
int n,m,k;
int e[1005][1005];
bool visit[1005];
int num = 0;
void dfs(int v){
	visit[v] = true;
	for(int i=1;i<=n;i++){
		if(!visit[i]&&e[v][i]){
			dfs(i);
		}
	}
}
void dfsTrave(int lost){
	num = 0;
	fill(visit,visit+1005,false);
	visit[lost] = true;
	for(int i=1;i<=n;i++){
		if(!visit[i]){
			dfs(i);
			num++;
		}
	} 
}
int main(){
	int m, k, a, b;
    scanf("%d%d%d", &n, &m, &k);
	for(int i=0;i<m;i++){
		scanf("%d%d", &a, &b);
		e[a][b] = e[b][a] = 1;
	}
	int lost;
    for(int i = 0; i < k; i++) {
        fill(visit, visit + 1005, false);
        scanf("%d", &lost);
        dfsTrave(lost);
        printf("%d\n", num - 1);
    }
}