AtCoder Grand Contest 008 D - K-th K 构造

最新推荐文章于 2021-07-12 18:15:59 发布

原创最新推荐文章于 2021-07-12 18:15:59 发布 · 408 阅读

0 ·

CC 4.0 BY-SA版权

构造专栏收录该内容

20 篇文章

订阅专栏

本文介绍了一种构造特定模式数组的算法，该算法要求输入的n个数组x[1..n]，输出一个长度为n*n的数组a，使得a中1到n每个数恰好出现n次，并且对于任意的i，从左到右第i个出现的i的下标为x[i]。文章详细介绍了算法的具体实现步骤，并给出了完整的C++代码。

题意

给出n个数组x[1..n]，要求构造一个长度为n*n的数组a，使得a中1到n每个数恰好出现n次，且对于任意的i，有从左到右第i个出现的i的下标为x[i]。
n<=500

分析

直接把1到n按照x从小到大排序，对于每个i，把前i个i填进去。
然后再从小到大把剩下的位置补完即可。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;

const int N=505;

int n,x[N],f[N],a[N*N+N];

bool cmp(int p,int q)
{
    return x[p]<x[q];
}

int main()
{
    scanf("%d",&n);
    for (int i=1;i<=n;i++) scanf("%d",&x[i]),f[i]=i;
    sort(f+1,f+n+1,cmp);
    int now=1;
    for (int i=1;i<=n;i++)
    {
        int w=f[i];a[x[w]]=w;
        for (int j=1;j<w;j++)
        {
            while (a[now]) now++;
            a[now]=w;
        }
        if (now>x[w]) {puts("No");return 0;}
    }
    for (int i=1;i<=n;i++)
    {
        int w=f[i];
        for (int j=1;j<=n-w;j++)
        {
            while (a[now]) now++;
            if (now<x[w]) {puts("No");return 0;}
            a[now]=w;
        }
    }
    puts("Yes");
    for (int i=1;i<=n*n;i++) printf("%d ",a[i]);
    return 0;
}