PAT甲题题解-1013. Battle Over Cities (25)-求联通分支个数

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <string>
#include <vector>
#include <cstring>
using namespace std;
const int maxn=1000+5;
int n,m,k;
int check[maxn]; //先标记哪些点会check，防止有重复的点做重复的工作
int ans[maxn]; //删掉节点i后剩余的连通分支数目
int vis[maxn]; //用于dfs时候的标记
struct Edge{
    int to;
    int next;
}edge[maxn*maxn];
int head[maxn];
int tot;

void init(){
    memset(head,-1,sizeof(head));
    tot=0;
}
void add(int u,int v){
    edge[tot].to=v;
    edge[tot].next=head[u];
    head[u]=tot++;
}

void dfs(int u){
    vis[u]=1;
    if(head[u]==-1)
        return;
    for(int k=head[u];k!=-1;k=edge[k].next){
        int v=edge[k].to;
        if(!vis[v]){
            dfs(v);
        }
    }
}
/*
对于要check的点，求删除后的联通分支数，存储在ans数组里
*/
void solve(){
    memset(ans,0,sizeof(ans));
    for(int i=1;i<=n;i++){
        //如果i是要被询问的
        if(check[i]){
            memset(vis,0,sizeof(vis));
            vis[i]=1;
            for(int j=1;j<=n;j++){
                //每次有没被访问过的节点，表明又是一个新的联通分支
                if(!vis[j] && j!=i){
                    dfs(j);
                    ans[i]++;
                }
            }
        }
    }
}
int main()
{
    int u,v;
    init();
    scanf("%d %d %d",&n,&m,&k);
    for(int i=0;i<m;i++){
        scanf("%d %d",&u,&v);
        add(u,v);
        add(v,u);
    }
    memset(check,0,sizeof(check));
    vector<int>query;
    for(int i=0;i<k;i++){
        scanf("%d",&u);
        check[u]=1;
        query.push_back(u);
    }
    solve();
    for(int i=0;i<query.size();i++){
        u=query[i];
        printf("%d\n",ans[u]-1);
    }
    return 0;
}