PAT A1013 Battle Over Cities [图的遍历]

题目描述

链接

给出n个城市之间有相互连接的m条道路，当删除一个城市和其连接的道路的时候，问其他几个剩余的城市至少要添加多少个路线才能让它们重新变为连通图

分析

• 要理解好题意，知道是让你求连通块的数目
• 然后就是如何求连通块数目：dfs！！！记得复习啊
• 对删除城市的处理：\(vis[a]=true\) ！！！！

#include <cstdio>
#include <algorithm>
using namespace std;
int v[1010][1010];
bool visit[1010];
int n;
void dfs(int node) {
    visit[node] = true;
    for(int i = 1; i <= n; i++) {
        if(visit[i] == false && v[node][i] == 1)
            dfs(i);
    }
}
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);
        v[a][b] = v[b][a] = 1;
    }
    for(int i = 0; i < k; i++) {
        fill(visit, visit + 1010, false);
        scanf("%d", &a);
        int cnt = 0;
        visit[a] = true;
        for(int j = 1; j <= n; j++) {
            if(visit[j] == false) {
                dfs(j);
                cnt++;
            }
        }
        printf("%d\n", cnt - 1);
    }
    return 0;
}

我的代码就是写复杂了！！！

#include<bits/stdc++.h>
using namespace std;

const int maxn = 1010;
bool G[maxn][maxn];
bool vis[maxn];
int n,m,k,st,ed,q;

void dfs(int p, int q){
    vis[p] = true;
    if(p == q) return;
    for(int i=1;i<=n;i++){
        if(i==q) continue;
        if(vis[i]==false && G[p][i]){
            dfs(i, q);
        }
    }
}

void solve(int q){
    int cnt = 0;
    memset(vis, 0, sizeof(vis));
    for(int i=1;i<=n;i++){
        if(vis[i]==false){
            dfs(i, q);
            cnt++;
        }
    }
    printf("%d\n", cnt-2);
}

int main(){
    cin>>n>>m>>k;
    memset(G,0,sizeof(G));
    for(int i=0;i<m;i++){
        cin>>st>>ed;
        G[st][ed] = G[ed][st] = 1;
    }
    for(int i=0;i<k;i++){
        cin>>q;
        solve(q);
    }


}