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