这道题的正解挺妙的。(标算2)
首先一个显然的性质:0,2之间的相对位置,1之间的相对位置是不变的
这个性质在考场上我想到了,但没有进一步推向正解,反而推了个奇奇怪怪的东西,似乎是标算1的错误实现,标算2更简洁明了。
可以开两个数组将0,2和1分开储存,用指针扫,加入数组
由于字典序最小,贪心让最小位尽量小,这样在两个数组指针所对应的权值中,我们势必要选择一个更小的
但是受到交换次数k的限制,并不是每一次最小的都可以选择,那么如何统计更小值放在答案数组最后一位的所需的交换次数?
答案就是max(0,i-len),i是指针位置,len是答案数组目前加到第几位。
若i-len<=0,说明这个数已经被其它的数所交换,被换到了指定位置,不需要再交换
反之需要i-len次额外交换操作使得更小值换到目标位置
每次算是否可行,若可以通过交换达到目的,将总次数减去这次需要交换的次数即可
如果我们因为操作次数不够选择了更大权值的位置,则其不需要交换。
实现起来特别简单。
#include <cstdio>
#include <cstring>
#include <algorithm>
#define int long long
using namespace std;
int i,j,k,m,n,o,p,l,s,t,p1,p2;
int a[1000005],b[1000005],c[1000005],pr[1000005];
void read(int &x)
{
char ch=getchar();x=0;
while (ch<'0'||ch>'9') ch=getchar();
while (ch>='0'&&ch<='9') x=(x<<3)+(x<<1)+ch-48,ch=getchar();
}
signed main()
{
freopen("mobiliarbus.in","r",stdin);
freopen("mobiliarbus.out","w",stdout);
read(n),read(m);
for (i=1;i<=n;i++) read(a[i]),a[i]==1?c[++c[0]]=i:b[++b[0]]=i;
pr[0]=1,p1=p2=1;
for (i=1;i<=n;i++)
{
if (p1>b[0]||p2>c[0]) break;
if (a[b[p1]]<a[c[p2]])
{
o=max(0ll,b[p1]-pr[0]);
if (o<=m) m-=o,pr[pr[0]++]=a[b[p1++]];
else pr[pr[0]++]=a[c[p2++]];
} else {
o=max(0ll,c[p2]-pr[0]);
if (o<=m) m-=o,pr[pr[0]++]=a[c[p2++]];
else pr[pr[0]++]=a[b[p1++]];
}
}
if (p1>b[0])
for (i=p2;i<=c[0];i++) pr[pr[0]++]=a[c[i]];
else if (p2>c[0])
for (i=p1;i<=b[0];i++) pr[pr[0]++]=a[b[i]];
for (i=1;i<=pr[0]-1;i++) printf("%d ",pr[i]);
return 0;
}