CF67B Restoration of the Permutation
分析:
题目告诉了我们一种规则由a数组变成b数组。如
A={5,1,4,2,3} ,k=2; 当我们求bi时,先到a数组找i=aj;看a1-aj中有几个数是满足ax<=i+k的,满足的数的个数即为bi的值。
求b1时,先找到1在a数组中得位置,得到1左边的数 5;有5>=1+2;所以b1=1;
求b2时,先找到2在a数组中得位置,得到1左边的数 5,1,4;
有5>=2+2; 4>=2+2 ; 所以b2=2;
以此类推。
当题目是给我们b数组,要我们求最小得字典序的a数组。我们可以想到这样一个事实,
在b中第一个0出现的位置,放在a[1],这时候放b中最小的满足条件的值为1的值放在a[2]中(在程序中实现时更新b中得值,使满足aj>=i+k的b的下标cur(cur=aj)中得值减一)。
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
int main()
{
int a[1005],b[1005];
int n,k;
while(scanf("%d%d",&n,&k)==2)
{
for(int i=1;i<=n;i++) scanf("%d",b+i);
for(int i=1;i<=n;i++)
{
int cur=1;
while(b[cur]!=0) cur++; //找最近的0;
a[i]=cur;
b[cur]--; //将放进了a中的数cur,变成b[cur]=-1,这样就标记了。
for(int j=1;j+k<=cur;j++) b[j]--; //更新b数组。
}
for(int i=1;i<=n;i++) printf(i!=n?"%d ":"%d\n",a[i]);
}
return 0;
}