PAT 1013. Battle Over Cities (25) 求连通分量个数

#include<cmath>
#include<stdio.h>
#include<algorithm>
#include<cstring>
#include<iostream>
#include<stack>
#include<vector>
#include<queue>
#include<string>
#include<map>
using namespace std;

/*************************
题意：给出一个无向图
求删去其中一个节点时，需要增加多少边才能使之相连
/************************
求解要点：
增加多少边实际上就是删去节点时，有多少连通分量
这里我们可以用dfs简单去做。
求联通分量个数方法：对每个点做dfs搜索，经过一个点就vis置1
搜完后，即为1个连通分量
接着对其他点进行同样dfs搜索
如果vis未被置为1，则为新的联通分量。
************************/

/***********************
笔记：
自己写代码前最好先想好特殊情况
*********************/
#define M 1005
vector<int> node[M]; 
int vis[M];

void dfs(int city){
	int i,nextc;
	for(i = 0;i < node[city].size(); i++){
		nextc = node[city][i];
		if(!vis[nextc]){
			vis[nextc] = 1;
			dfs(nextc);
		}
	}

}

int main()
{
	int n,m,k;
	scanf("%d%d%d",&n,&m,&k);

	int s,e;
	while(m--){
		scanf("%d%d",&s,&e);
		node[s].push_back(e);
		node[e].push_back(s);
	}
	int c,i,city;
	int sum;
	while(k--){
		scanf("%d",&c);
		memset(vis,0,sizeof(vis));
		sum = 0;
		vis[c]=1;//这个城市不能走了
		for(i=1;i<=n;i++){
			if(!vis[i]){  //对一个点进行dfs搜索，经过的点置1
				sum++;    //若这个点没被置1，说明存在一个连通分量
				vis[i]=1;
				dfs(i);  
			}
		}
		printf("%d\n",sum-1);
	
	}
	return 0;
}