牛客题目 - 白色相簿（并查集）

最新推荐文章于 2022-06-17 22:12:18 发布

__Wedream__

最新推荐文章于 2022-06-17 22:12:18 发布

阅读量469

点赞数 1

CC 4.0 BY-SA版权

分类专栏：算法之路文章标签：数据结构 c++ 程序设计

本文链接：https://blog.csdn.net/weixin_43800577/article/details/107380839

算法之路专栏收录该内容

24 篇文章

订阅专栏

本文深入解析了牛牛的Fib序列问题，这是一个典型的并查集题目，通过并查集算法判断一组关系中是否存在三人及以上的不稳定关系。文章详细介绍了并查集的实现方式，包括寻找根节点和合并元素的方法，并提供了完整的代码示例。

牛牛的Fib序列

Problem Description

链接：https://ac.nowcoder.com/acm/problem/20793
来源：牛客网

众所周知，一个人或两个人之间的关系是最为稳定的，如果产生三个人以上的关系，则会变得不稳定
容易产生诸如“北原春希、冬马和纱、小木曾雪菜”这样的悲剧爱情故事
那么，为了避免这样的事情发生，我们需要判断现有的n个人里，是否存在三人及以上的关系
例如：A和B之间存在关系，B和C之间存在关系，则ABC之间即构成了一个三人关系，此时是不稳定的

Input Description

第一行两个整数n和m，表示一共有n个人，m对关系
接下来m行，每行两个整数x和y，表示第x个人和第y个人有关系
注：x和y可能相等，已经出现过的关系也可能重复出现
0≤n,m≤10^4

Output Description

输出仅一行，如果n个人中存在3人及以上的不稳定关系，则输出“Error”
否则，如果n人中仅存在1人或2人之间的稳定关系，则输出“Nice”

Example1

输入

输出

Nice

Solution

很经典的并查集题目，可以直接套用并查集模板即可。
若第x个人和第y个人有关系，则合并x元素和y元素。
最后遍历所有元素，用一个数组记录每个元素孩子的数目。
若大于或等于3个孩子，则输出“Error”；反之输出“Nice”。

并查集相关详情可参考博客并查集详解 ——图文解说,简单易懂

Code

#include<bits/stdc++.h>
using namespace std;

const int N=1e4+100;
int f[N];
int n,m;
int sum[N];

int find(int x)//寻找x的根节点
{
    return x==f[x]?x:f[x]=find(f[x]);
}

void Union(int a,int b)//合并两个元素
{
    int fa=find(a);
    int fb=find(b);
    if(fa!=fb)
        f[fa]=fb;
}

int main()
{
#ifdef LOCAL
    freopen("in.txt","r",stdin);
#endif
    scanf("%d%d",&n,&m);

    for(int i=1; i<=n; i++) //初始化
        f[i]=i;

    int u,v;
    while(m--)
    {
        scanf("%d%d",&u,&v);
        if(u==v)continue;
        Union(u,v);
    }

    int fa;
    for(int i=1; i<=n; i++)
    {
        fa=find(i);
        sum[fa]++;
        if(sum[i]>=3)
        {
            puts("Error");
            return 0;
        }
    }

    puts("Nice");

    return 0;
}