并查集求集合数

最新推荐文章于 2023-01-12 17:42:33 发布

stormjing7

最新推荐文章于 2023-01-12 17:42:33 发布

阅读量846

点赞数

分类专栏：解题报告比赛并查集

本文链接：https://blog.csdn.net/GodJing007/article/details/89381042

版权

解题报告同时被 3 个专栏收录

178 篇文章 0 订阅

订阅专栏

比赛

46 篇文章 0 订阅

订阅专栏

并查集

3 篇文章 0 订阅

订阅专栏

题目链接

在这里插入图片描述

题意：先给你城市数和道路数，告诉你那些城市连通的，之后有q次询问，每次询问销毁一个城市，问你还剩下几个集合数，即连通快。

分析：倒着建立并查集，题目依次销毁一个城市，我们就从后面每次加入一个城市，最开始要询问的城市都不加入。最后倒着输出结果。

#include <bits/stdc++.h>
#define INF 0x3f3f3f3f
#define d(x) cout << (x) << endl
#pragma GCC diagnostic error "-std=c++11"
using namespace std;
typedef long long ll;
const int mod = 1e9 + 7;
const int N = 1e4 + 10;

int n, m, q;
int co = 0; 	
int cnt = 0;		//集合数

vector<int> v[N];	//城市关系用邻接表表示
int pre[N];
int city[N];
int ans[N];		
bool vis[N];	//还没加入的城市置为1

int find(int x)
{
    return pre[x]==x?x:pre[x]=find(pre[x]);
}

void init()		//最开始要询问的城市都不加入，此时的集合数为最后一个结果。
{
    for(int i = 1; i <= n; i++){
        if(!vis[i]){
            for(auto x : v[i]){
                if(!vis[x]){
                    pre[find(x)] = find(i);
                }
            }
        }
    }
    for(int i = 1; i <= n; i++){
        if(!vis[i] && find(i) == i)
            cnt++;
    }
    ans[co++] = cnt;
}

int main()
{
    scanf("%d%d", &n, &m);
    int x, y;
    for (int i = 0; i < m; i++)
    {
        scanf("%d%d", &x, &y);
        v[x].push_back(y);
        v[y].push_back(x);
    }
    for(int i = 1; i <= n; i++){
        pre[i] = i;
    }
    scanf("%d", &q);
    for (int i = 0; i < q; i++)
    {
        scanf("%d", &city[i]);
        vis[city[i]] = 1;
    }
    init();
    for (int i = q - 1; i > 0; i--)
    {
        vis[city[i]] = 0;
        cnt++;      //每次都有一座城市进来
        for(auto x : v[city[i]]){
            if(!vis[x] && find(x) != find(city[i])){
                cnt--;              //两个集合并在一起，就少了一个集合
                pre[find(x)] = find(city[i]);
            }
        }
        ans[co++] = cnt;
    }
    for (int i = co - 1; i >= 0; i--){		//最后一个输出回车
        printf("%d%c", ans[i], " \n"[i == 0]);	
    }
    return 0;
}