PAT 甲级 1013 Battle Over Cities DFS搜索

地址 https://pintia.cn/problem-sets/994805342720868352/problems/994805500414115840

题目的大意是

一个双向连通图，如果我们去掉其中一个点，要求输出还需要添加多少条路线才能满足剩余点全部连通的要求.

输入格式：

第一行 n,m,k.    n表示有n个点(编号1~n)  m表示有m条路线  k表示k个查询，
接下来m行 每行两个整数 a b ，表示a和b之间有双向路径连接。

最后一行 k个数字  每个数字表示点id号，查询某个点去掉后，需要连接几条路线才能满足剩余点全部连通的要求.

输出格式：

输出k行 每行一个数字 表示去掉某点后，需要添加多少条路线才能满足剩余点全部连通的要求

解答

本题目本来是考核并查集的使用
但是由于数据规模适中，使用深度优先搜索DFS 也是OK的
去掉题意中被占领的那座城市 看看进行几次DFS搜索能遍历完全部图中的城市节点
那么就是答案
图示

#include <iostream>
#include <vector>
#include <memory.h>


using namespace std;


const int N = 1010;

vector<int> graph[N];
int vis[N];
int lost,ans;

int n,m,k;
int a,b;

void dfs(int x){
    vis[x]=1;
    int size = graph[x].size();
    for(int i =0;i < size;i++){
        int p = graph[x][i];
        if(vis[p]==0 && p!=lost){
            dfs(p);   
        }
    }
}


void solve(){
    ans =0;
    memset(vis,0,sizeof vis);
    for(int i = 1;i<=n;i++){
        if(i!=lost && vis[i] ==0){
            ans++; dfs(i);    
        }
    }

    printf("%d\n",ans-1);
    return ;
}

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

    for(int i = 1; i <= m;i++){
        scanf("%d%d",&a,&b);
        graph[a].push_back(b);
        graph[b].push_back(a);
    }

    for(int i =1;i<=k;i++){
        scanf("%d",&lost);
        solve();
    }

    return 0;
}