PAT 1013. Battle Over Cities （联通块数量）

版权声明：本文为博主原创文章，转载请注明出处。 https://blog.csdn.net/flyawayl/article/details/79308336

分析

如果某个城市被敌人占领，就相当于从图上将这个点以及与这个点相连的所有边删除，那么剩下的图可能被分成几个规模更小的图，也就是说现在有几个联通块。这样一想来就简单了，直接求联通块数量就好了，然后用几条边把所有联通块连接起来就能让所有城市联通！！

AC代码

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <math.h>
#include <string>
#include <stack>
#include <queue>
#include <set>
#include <map>
#include <vector>
#include <algorithm>
#include <iostream>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
#define ms(s) memset(s, 0, sizeof(s))
#define INF 0x3f3f3f3f
const int maxn = 1000 + 5;

bool G[maxn][maxn];
bool vis[maxn];
int n, m, k;

void dfs(int u, int enemy) {
    vis[u] = true;
    for(int i = 1; i <= n; i++)
        if(i != enemy && G[u][i] && !vis[i]) dfs(i, enemy);
}

int check(int enemy) {
    ms(vis);
    int num = 0;
    for(int i = 1; i <= n; i++) {
        if(i != enemy && !vis[i]) {
            num++;
            dfs(i, enemy);
        }
    }
    return num;
}

int main() {
    while(scanf("%d%d%d", &n, &m, &k) == 3) {
        ms(G);
        int u, v;
        for(int i = 0; i < m; i++) {
            scanf("%d%d", &u, &v);
            G[u][v] =  G[v][u] = true;
        }
        int enemy;
        for(int i = 1; i <= k; i++) {
            scanf("%d", &enemy);
            int blocks = check(enemy);
            printf("%d\n", blocks - 1);
        }
    }
    return 0;
}

Tips：就算有多个点被删除也能用同样的方法解决。

---

如有不当之处欢迎指出！