AGC008D K-th K

题意简述：给你一个长度为\(N\)的整数序列\(x\)，请判断是否存在一个满足下列条件的整数序列\(a\)，如果存在，请构造一种方案。

1.\(a\)的长度为\(N^2\)并且满足数字\(1,2,3,\cdots,N\)都各出现恰好\(N\)次

2.对于\(1<=i<=N\)，数字\(i\)在\(a\)中第\(i\)次出现的位置是\(x_i\)

我他么写这种傻逼题写了1.5h，cao

非常简单，我们可以考虑贪心，将二元组\((x,i)\)按\(x\)排序，然后\(x_i\)放\(i\)，前面暴力放最前面空的\(i-1\)个，再倒着做一遍就可以了

时间复杂度\(O(N^3)\)

#include<cstdio>
#include<algorithm>
using namespace std;
const int N=510;
struct fk{int x,id;}q[N];
int n,a[N*N],cnt[N],flag;
bool cmp(fk a,fk b){return a.x<b.x;}
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        scanf("%d",&q[i].x),q[i].id=i;
    sort(q+1,q+n+1,cmp);
    for(int i=1;i<=n;i++){
        int id=q[i].id;
        a[q[i].x]=id;++cnt[id];
        for(int j=1;j<q[i].x;j++){
            if(cnt[id]==id)break;
            if(!a[j])a[j]=id,++cnt[id];
        }
        if(cnt[id]<id)flag=1;
    }
    if(flag){puts("No");return 0;}
    flag=0;
    for(int i=n;i>=1;i--){
        int id=q[i].id;
        for(int j=n*n;j>q[i].x;j--){
            if(cnt[id]==n)break;
            if(!a[j])a[j]=id,++cnt[id];
        }
        if(cnt[id]<n)flag=1;
    }
    if(flag){puts("No");return 0;}
    puts("Yes");
    for(int i=1;i<=n*n;i++)printf("%d ",a[i]);
    puts("");
}