【PAT甲级】1013 Battle Over Cities (25 分)【并查集】

【PAT甲级】1013 Battle Over Cities (25 分)

---

思路

• 如图有6个节点，5条边，如果删除6号节点，那么图中将只有三个连通块，分别是【1】【4】【2,3,5】
• cnt初始化为n-1，表示有1 ，2 ，3 ，4 ，5五个互不连通的节点
• 遍历所有的边，如果说该边连接的两个节点不包含需要删除的那个节点，那么使用并查集操作进行判断：如果在并查集中，a，b不连通，则执行合并操作，同时使cnt-1（注意理解）
• 执行完上一步之后，cnt为连通块的数量，因此要使它们连通只需添加cnt-1条边，输出cnt-1即可

---

代码

#include <iostream>

using namespace std;
const int N=1e3+10;
const int M=1e6+10;
int p[N];
int n,k,m;

struct edge
{
    int a,b;
}e[M];

int find(int x)
{
    if(x!=p[x]) p[x]=find(p[x]);
    return p[x];
}

int main()
{
    cin>>n>>k>>m;//城市总数，高速公路总数，重点关注城市数量。
   
    for(int i=0;i<k;i++) cin>>e[i].a>>e[i].b;//读入k条边的信息
    
    while(m--)
    {
        int x;cin>>x;//x是要删除的点的值
        for(int i=1;i<=n;i++) p[i]=i;//n个互相独立的集合
        int cnt=n-1;//cnt保存答案，初始为总结点数-1，也即是n-1
        for(int i=0;i<k;i++)//遍历所有的边
        {
            int a,b;
            a=e[i].a,b=e[i].b;
            if(a!=x&&b!=x) 
            {
                if(find(a)!=find(b))//如果a，b不连通，则执行合并操作
                {
                    p[find(a)]=find(b);//合并操作
                    cnt--;//每合并一次，相当于修建一条高速公路
                }
            }
        }
        //最终保存的是连通块的个数，减一就是最终的结果
        cout<<cnt-1<<endl;
    }
    return 0;
}