1118 Birds in Forest（25 分）【并查集】

最新推荐文章于 2020-07-07 17:59:07 发布

QQN1996

最新推荐文章于 2020-07-07 17:59:07 发布

阅读量310

点赞数

分类专栏： PAT 图并查集

本文链接：https://blog.csdn.net/QQN19961117/article/details/82389017

版权

该博客探讨了一道编程问题，涉及同一张图中鸟群分布的分析。通过使用并查集数据结构，求解鸟儿们所在的树木数量，并能判断任意两只鸟是否属于同一棵树。内容详述了如何运用并查集解决此类问题。

摘要由CSDN通过智能技术生成

题意：同一张图中的bird在同一棵tree上，问一共有几棵tree，并判断任意两只bird是否在同一棵tree上。

#include <bits/stdc++.h>
using namespace std;
#define INF 0x7FFFFF

int fa[10005];

void Init()
{
    for(int i=1;i<10005;i++)
        fa[i]=i;
}

int Find(int x)
{
    if(x==fa[x])
        return fa[x];
    else
        return fa[x]=Find(fa[x]);  //必须进行路径压缩，不然会超时
}

void Union(int x,int y)
{
    int xx=Find(x);
    int yy=Find(y);
    if(xx!=yy)
        fa[xx]=yy;
}

int main()
{
    int n;
    cin>>n;

    set<int> s;
    Init();

    while(n--){
        int k;
        cin>>k;
        int x;
        cin>>x;
        s.insert(x);
        for(int i=1;i<k;i++){
            int y;
            cin>>y;
            Union(x,y);
            s.insert(y);
            x=y;
        }
    }

    int sum=0;//bird数量
    sum=s.size()