版本2
牛客网测试点
需要对边为0的加个特判
直接用深搜加标记就可以统计出结果。
#include <iostream>
#include <string>
#include <vector>
#include <cctype>
#include <algorithm>
using namespace std;
int n, m, k;
const int maxn = 1010;
vector<int> G[maxn];
void dfs(int v, vector<bool> &vis){
vis[v] = true;
for(int j = 0; j < G[v].size(); j++){
int u = G[v][j];
if(!vis[u]) dfs(u, vis);
}
}
int main(){
int u, v;
cin >> n >> m >>k;
for(int i = 0; i < m; i++){
cin >> u >> v;
G[u].push_back(v);
G[v].push_back(u); //无向图
}
while(k--){
cin >> u;
if(m == 0){
cout<<0<<endl;
continue;
}
vector<bool> vis(n+1);
vis[u] = true;
int cnt = -1;
for(int i = 1; i <= n; i++){
if(!vis[i]){
dfs(i, vis);
cnt++;
}
}
cout<<cnt<<endl;
}
return 0;
}
开始的时候题目理解错了
题目是要求剩下的城市连通,所以需要当前城市屏蔽,然后计算剩余城市的连通分量,求得的结果减去1就是答案
本题我使用的BFS处理
版本1
#include<cstdio>
#include<vector>
#include<cstring>
#include<queue>
#include<algorithm>
using namespace std;
const int maxn=1005;
int vis[maxn];
int n,m,k;
vector<int> G[maxn];
int bfs(int s){
queue<int> q;
memset(vis,0,sizeof(vis));
int cnt=0;
for(int i=1;i<=n;i++){
if(i!=s && vis[i]==0){
q.push(i);
vis[i]=1;
while(!q.empty()){
int tmp = q.front();
q.pop();
for(int j=0;j<G[tmp].size();j++){
int v=G[tmp][j];
if(v != s && vis[v]==0){
vis[v]=1;
q.push(v);
}
}
}
cnt++;
}
}
return cnt;
}
int main(){
scanf("%d%d%d",&n,&m,&k);
int u,v;
while(m--){
scanf("%d%d",&u,&v);
G[u].push_back(v);
G[v].push_back(u);
}
while(k--){
scanf("%d",&u);
printf("%d\n",bfs(u)-1);
}
return 0;
}