1013 Battle Over Cities - 并查集

思路:

简单并查集,就是我们不考虑一个城市的时候,可以排除它用并查集,看看最后有几堆,那么ans就是堆数-1,(要把这几堆连起来)

犯傻的地方:

(1)qwq我开始看题还在想,用并查集怎么把取消的那个城市相关操作取消掉qwq,我是个傻子吗???离线处理不就好了qwq

(2)求堆数的时候还开了个vis数组,标记每个城市的父节点???若这个点没被标记,则ans++???

直接看看有几个点他的父节点就是自己不就知道有几堆了嘛qwq

代码如下:

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<string>
#include<cstring>
#include<queue>
#include<cmath>
#include<set>
#include<map>
#define ll long long
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
using namespace std;
const int N=1010;
int n,m,a[N*N],b[N*N],pre[N];

int f(int x){
    if(pre[x]==x)return x;
    pre[x]=f(pre[x]);
    return pre[x];
}

void link(int x,int y){
    int r1=f(x),r2=f(y);
    if(r1!=r2){
        pre[r2]=r1;
    }
}
void solve(int city){
    for(int i=1;i<=n;i++){
        pre[i]=i;
    }
    for(int i=0;i<m;i++){
        if(a[i]==city||b[i]==city)continue;
        link(a[i],b[i]);
    }
    int ans=0;
    for(int i=1;i<=n;i++){
        if(i==city)continue;
        int ri=f(i);
        if(ri==i)ans++;
    }
    printf("%d\n",ans-1);
}

int main(){
    int k;
    scanf("%d%d%d",&n,&m,&k);
    for(int i=0;i<m;i++){
        scanf("%d%d",&a[i],&b[i]);
    }
    int x;
    for(int i=1;i<=k;i++){
        scanf("%d",&x);
        solve(x);
    }
}

猜你喜欢

转载自blog.csdn.net/m0_37579232/article/details/83096594