Codeforces 1256D Binary String Minimizing

思路:

1.每输入进来一个0,看它是否能移动到最前端(前面不存在字符‘1’即为最前端),计算所需要的移动次数,和k做比较;
2.k够用就输出一个0;不够的话计算最多能移动到哪,输出相应数量的1,再输出这个0,再输出剩下的1,然后既然k用完了,接下来输入什么就输出什么;
3.输入输出的数据量较多,可以考虑优化一下cin和cout,或者直接用scanf和getchar;

代码:

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
#define rp(i,n) for(int i=0;i<n;i++)
#define rpn(i,n) for(int i=1;i<=n;i++)
const int MAX_N=1000005;
int main(){
	int q;
	scanf("%d",&q);
	rp(i,q){
		ll n,k;
		scanf("%lld %lld",&n,&k);
		getchar();
		ll ans=0;//统计当前0的个数 
		bool flag=false;//k是否被用完 
		rp(j,n){
			char c=getchar();
			if(flag) putchar(c);
			else if(c=='0'){
				ll d=j-ans;//移到最前面所需要的次数 
				if(k<=d){//k被用完了,不够0被移到最前面 
					ll pos=j-k;
					for(pos=ans;pos<j-k;pos++) putchar('1');
					putchar('0');
					for(pos++;pos<j+1;pos++) putchar('1');
					flag=true;					
				}else putchar('0');	//k足够此次的0被移到最前端					
				ans++;
				k-=d; 
			}
		}	
		if(k>0) for(int pos=ans;pos<n;pos++) putchar('1');//k没有被用完 
		putchar('\n');
	}
	return 0;
}
posted @ 2019-11-05 15:11  YuhanのBlog  阅读(76)  评论(0编辑  收藏  举报