[agc008d]K-th K

前言

简单构造，一开始看错题瞎写了一发。

题目大意

给了一个长度为n的序列x。
请构造一个长度为n*n的序列a，只包含1~n这n个元素，每个元素出现n次，且对于元素i，a的第xi个位置是元素i，且前xi个位置恰好有i个i。

构造

显然可以正反各贪心一遍，不懂就看程序，正确性显然。
瞎填以后还可以n^3验证。

#include<cstdio>
#include<algorithm>
#define fo(i,a,b) for(i=a;i<=b;i++)
#define fd(i,a,b) for(i=a;i>=b;i--)
using namespace std;
const int maxn=500+10;
bool bz[maxn];
int cnt[maxn];
int x[maxn],b[maxn],a[maxn*maxn];
int i,j,k,l,t,n,m,tot,top;
bool czy;
int main(){
    scanf("%d",&n);
    fo(i,1,n){
        scanf("%d",&t);
        a[x[i]=t]=i;
    }
    fo(i,1,n*n) 
        if (a[i]) b[++top]=a[i];
    fo(i,1,n) cnt[i]=i-1;
    fo(i,1,n*n)
        if (!a[i]){
            fo(j,1,n)
                if (cnt[b[j]]){
                    a[i]=b[j];
                    cnt[b[j]]--;
                    break;
                }
        }
    fo(i,1,n) cnt[i]=n-i;
    fd(i,n*n,1)
        if (!a[i]){
            fd(j,n,1)
                if (cnt[b[j]]){
                    a[i]=b[j];
                    cnt[b[j]]--;
                    break;
                }
        }
    czy=1;
    fo(i,1,n){
        tot=0;
        fo(j,1,n*n)
            if (a[j]==i){
                tot++;
                if (tot==i) break;
            }
        if (j!=x[i]){
            czy=0;
            break;
        }
    }
    if (!czy) printf("No\n");
    else{
        printf("Yes\n");
        fo(i,1,n*n) printf("%d ",a[i]);
    }
}