CodeForces - 28B pSort【并查集】

最新推荐文章于 2022-03-23 15:02:31 发布

invoKer-

最新推荐文章于 2022-03-23 15:02:31 发布

阅读量264

点赞数

CC 4.0 BY-SA版权

分类专栏：并查集

本文链接：https://blog.csdn.net/inv00ker/article/details/113011473

并查集专栏收录该内容

7 篇文章

订阅专栏

该博客介绍了一种使用并查集数据结构解决连通性问题的方法。通过读取序列和距离数据，将节点连接起来，并检查序列中的元素是否属于同一连通分量，从而判断序列是否有效。程序通过初始化、路径压缩和合并操作来实现并查集，最后输出判断结果。

题目链接：https://codeforces.com/contest/28/problem/B

#include <iostream>
using namespace std;
static const int MAXN=100+10;
int p[MAXN];
int seq[MAXN],d[MAXN];
int n;
void init() { for(int i=1;i<=n;i++) p[i]=i; }
int find(int x)
{
    if(p[x]!=x) p[x]=find(p[x]);
    return p[x];
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%d",&seq[i]);
    init();
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&d[i]);
        if(i-d[i]>=1)
        {
            int u=i;
            int v=i-d[i];
            int pu=find(u),pv=find(v);
            if(pu!=pv) p[pu]=pv;
        }
        if(i+d[i]<=n)
        {
            int u=i;
            int v=i+d[i];
            int pu=find(u),pv=find(v);
            if(pu!=pv) p[pu]=pv;
        }
    }
    bool flag=true;
    for(int i=1;i<=n;i++)
    {
        int u=i,v=seq[i];
        int pu=find(u),pv=find(v);
        if(pu!=pv) flag=false;
    }
    puts(flag?"YES":"NO");
    return 0;
}