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

题意

给出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;
}