pat 甲级 A1013 Battle Over Cities (25分)

题目链接：

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

 

问题分析：

并查集+树状数组。

N个城市之间由M条道路连接，给出K个查询，每次查询给出可能被敌方占领的城市编号。要求你算出相应在该城市被占领后，至少需要修多少条道路使得其变成连通的图。即，要求你算出图中有多少个连通分量。可以用并查集的方法求解。

 

参考代码：

#include <cstdio>
#include <vector>
using namespace std;
const int maxn = 1010;
vector <int> G[maxn];
int father[maxn], n, m, k;
bool vis[maxn] = {false};

int findfather(int x){     //压缩路径，查找节点x的根节点，然后直接将节点x连到根节点上。
    return father[x] == x ? x :father[x] = findfather(father[x]);
}
void setUnion(int a, int b){   //将节点a,节点b合并。
    int f1 = findfather(a);
    int f2 = findfather(b);
    father[f2] = f1;
}

int main(){
    scanf("%d %d %d", &n, &m, &k);
    for(int i = 0; i < m; i++){    //得到图的连接矩阵。
        int a, b;
        scanf("%d %d", &a, &b);
        G[a].push_back(b);
        G[b].push_back(a);
    }
    int currentPoint;
    for(int query = 0; query < k; query++){ //k次查询
        scanf("%d",&currentPoint);
        for(int i = 1; i <= n; i++){
            father[i] = i;
            vis[i] = false;
        }
        for(int i = 1; i <= n; i++){    //遍历所有节点。
            for(int j = 0; j < G[i].size(); j++){   
                if(i == currentPoint || G[i][j] == currentPoint) continue;  //i , G[i][j] 
                                                              //     表示边i-j的两个端点。
                setUnion(i, G[i][j]);
            }
        }
        int block = 0; //表示连通分量的个数。
        for(int i = 1; i <= n; i++){
            if(i == currentPoint) continue;
            if(!vis[father[i]]){   //father[i]未访问过
                block++;
                vis[father[i]] = true;
            }
        }
        printf("%d\n", block - 1);
    }
    return 0;
}