PAT1013 Battle Over Cities

题目描述
考点：并查集
思路：利用邻接矩阵存储图，然后城市被攻占抹去所有与他相邻的边，之后遍历这个图建立并查集，最后查询有几个不同的集合，数目-1就是答案。
注意点：城市被攻占的时候，最后统计集合数目不能包括这个城市

#include<iostream>
using namespace std;
const int N = 1010;
int g[N][N];
int p[N];
int n,m,k;
int find(int x){
    if(x != p[x]) p[x] = find(p[x]);
    return p[x];
}
void query(int q){
    for(int i = 1;i <= n;i++){
        p[i] = i;
    }  
    for(int i = 1;i <= n;i++){
        for(int j = 1;j <= n;j++){
            if(g[i][j] == 1 && j != q && i != q){//不能等于q，因为被攻占了
                int pi = find(i);
                int pj = find(j);
                if(pi != pj)
                    p[i] = pj;
            }
        }
    }
    int cnt = 0;
    for(int i = 1;i <= n;i++) if(p[i] == i && i != q) cnt++;
    cout<<cnt-1<<endl;
}
int main(){
    cin>>n>>m>>k;
    while(m--){
        int a,b;
        cin>>a>>b;
        g[a][b] = g[b][a] = 1;
    }
    while(k--){
        int q;
        cin>>q;
        query(q);
    }
    return 0;
}