P6121 [USACO16OPEN]Closing the Farm G——并查集习题

最新推荐文章于 2024-05-20 09:37:58 发布

Annabel_CM

最新推荐文章于 2024-05-20 09:37:58 发布

阅读量368

点赞数 2

分类专栏：蓝桥杯题解

本文链接：https://blog.csdn.net/Annabel_CM/article/details/108185781

版权

蓝桥杯题解专栏收录该内容

25 篇文章 4 订阅

订阅专栏

在这里插入图片描述

解题思路：
题目要求：关闭某个谷仓之前，问这个谷仓是否为全联通
突破：将删点转化为加点
正序“删除每个点”前的状态等同于，倒序“增加每个点”后的状态

#include<iostream>
#include<vector>
#include<cstdio>
using namespace std;
const int N = 200010;
vector<int> e[N];//存储每个节点的**所有边信息**
int p[N], del[N];//p记录父节点，del记录删除顺序
int blocks;//存储当前连通块的数量
bool ok[N];//记录增加每个节点是否连通
int find(int x)
{
    if(p[x] != x) p[x] = find(p[x]);
    return p[x];
}
int merge(int x, int y)//是否需要合并集合，若两个在同一集合，不用合并
{
    if(find(x) == find(y)) return 0;
    p[find(x)] = find(y);
    return 1;
}
int main()
{
    int n, m;
    cin >> n >> m;
    int a, b;
    for(int i = 1; i <= m; i++)//记录双向连通
    {
        cin >> a >> b;
        e[a].push_back(b);//a和b连通
        e[b].push_back(a);//b和a连通
    }
    for(int i = 1; i <= n; i++) cin >> del[i];
    for(int i = n; i >= 1; i--)//倒序
    {
        int x = del[i];
        p[x] = x;//自成集合
        blocks++;//单个连通块增加
        for(int j = 0; j < e[x].size(); j++)
        {
            int y = e[x][j];//读出相连的节点
            if(!p[y]) continue;//如果该节点还没添加
            blocks -= merge(x, y);//x,y节点合并（-1）
        }
        if(blocks <= 1) ok[i] = true;
    }
    for(int i = 1; i <= n; i++)
    {
        if(ok[i] == true) puts("YES");
        else puts("NO");
    }
    return 0;
}