PAT甲级 - 1013 Battle Over Cities(删点判连通块数量)

Link

题意：给你一张地图，n个城市、m条路，现在问你毁掉某个城市相连的所有路后，最少还需要修建多少条路才能使得剩下的所有城市联通。一个k组测试询问。

思路：相当于删除联通块中某个点剩下联通块个数-1的问题。思路1：并查集，思路二：bfs遍历。需要注意的是上限1e3个点，那么最多有1e6个边，容量不够大最后一个测试点无法通过。

并查集代码：

#include<bits/stdc++.h>
#define int long long
#define endl '\n'
using namespace std;
const int N = 1e6+10;

int n, m, k, ans;
int f[N];

int find(int x){
    return f[x]==x ? x : f[x]=find(f[x]);
}

void join(int x, int y){
    int a = find(x), b = find(y);
    if(a != b){
        f[a] = b;
        ans --;  // 在建边的时候就要判断，而不能在最后遍历判断祖先节点的种数
    }
}

struct node{
    int a, b;
} g[N];

signed main()
{
    cin >> n >> m >> k;
    for(int i = 1; i <= m; i ++){
        cin >> g[i].a >> g[i].b;
    }

    while(k--){
        int x; cin >> x;
        for(int j = 1; j <= n; j ++) f[j] = j;
        ans = n-1;
        for(int j = 1; j <= m; j ++){
            if(g[j].a==x||g[j].b==x) continue;
            join(g[j].a, g[j].b);
//            cout << "a: " << g[j].a << " b: " << g[j].b << endl;
        }
        cout << ans-1 << endl;
    }


    return 0;
}

dfs代码：

#include<bits/stdc++.h>
#define int long long
#define endl '\n'
using namespace std;
const int N = 1005;

int n, m, k, ans;
bool vis[N];
int g[N][N];

void dfs(int x){  // dfs跑过所有能到达的点
    vis[x] = 1;
    for(int i = 1; i <= n; i ++){
        if(!vis[i]&&g[x][i] == 1){
            dfs(i);
        }
    }
}

signed main()
{
    cin >> n >> m >> k;
    for(int i = 1; i <= m; i ++){
        int a, b; cin >> a >> b;
        g[a][b] = g[b][a] = 1;
    }
    while(k--){
        int x; cin >> x;
        memset(vis, 0, sizeof(vis));
        vis[x] = 1; ans = 0;
        for(int i = 1; i <= n; i ++){  // 每个未访问过的结点进行dfs
            if(!vis[i]){
                dfs(i);
                ans ++;
            }
        }
        cout << ans-1 << endl;
    }

    return 0;
}