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("");
}