本题题意:给定一个长度为n的二进制串,然后可以对序列中的0进行k次前后交换,使得0尽量位于1的前面。
——————————————————
简单分析发现一个0向前交换到的位置分为两种情况:
- k步之内移动到所有1以及一串0后面(第一个0的话前面没有0)
- k步之内无法移动到所有1前面
而移动的步数消耗与剩余步数,序列前0的数量,0所处位置与第一个1的距离有
关。设0所处位置为i,与第一个1的距离为d,序列前0 的数量及这一串0的最后位
置为cur,剩余步数为k1。
则d=i-cur,如果:
1.k1>=d,k1=k1-d,cur=cur+1。
2.k1<d,移动到的位置pos=i-k1,k1=0。
即:
if(k1+cur>=i){
k1-=i;
k1+=cir;
cur++;
}
else{
pos=i-k;
k1=0;
}
之后就是0移动的交换,如果按一般的一位位交换时间消耗大,会超时,于是分析对于一个序列:
00111111011101
要移动的加粗的0向前交换的都是1,就可以直接将要交换到位置赋值为0,原位置
赋值为1即可。
00111111011101
00011111111101
代码:
if(k+cur>=i){
k-=i;
k+=cur;
a[i]='1';
a[cur]='0';
cur++;
}
else{
long long pos=i-k;
k=0;
a[i]='1';
a[pos]='0';
}
这样就可以实现移动的基本操作。
于是便可以从序列头开始贪心遍历,如果元素为0执行移动操作即可。
#include<stdio.h>
#include<string.h>
#include<iostream>
using namespace std;
char a[1000005];
int main(){
int q,n,cur;
long long k;
scanf("%d",&q);
while(q--){
cur=0;
scanf("%d",&n);
cin>>k;
for(int i=0;i<=n;i++) a[i]=0;
getchar();
scanf("%s",a);
for(int i=0;i<n;i++){
if(!k) break;
if(a[i]=='0'){
if(k+cur>=i){
k-=i;
k+=cur;
a[i]='1';
a[cur]='0';
cur++;
}
else{
long long pos=i-k;
k=0;
a[i]='1';
a[pos]='0';
}
}
}
for(int i=0;i<n;i++) printf("%c",a[i]);
printf("\n");
}
return 0;
}
做题小结:注意数据范围