2023“钉耙编程”中国大学生算法设计超级联赛（2）Binary Number

2023“钉耙编程”中国大学生算法设计超级联赛（2）Binary Number

题目大意

给定一个二进制数，按每一位记为$s_1\sim s_n$（$s_1$是最高位，$s_n$是最低位）。你需要进行恰好$k$次操作：

• 任意选择一个区间$[l,r]$，将$s_l,s_{l+1},\dots ,s_r$翻转

也就是说，对于所有的$i\in [l,r]$，如果$s_i$为$0$，则将$s_i$变为$1$；如果$s_i$为$1$，则将$s_i$变为$0$。

求进行了$k$次操作以后，最大的结果二进制数是多少。

有$t$组数据。

$1\le t\le 6\times 10^4,1\le n\le 10^5,0\le k\le 10^{18}$

$s_1=1$，对于所有$i\in [2,n]$， s i ∈ { 0 , 1 } s_i\in\{0,1\} si​∈{0,1}

数据保证$\sum n\le 2.5\times 10^6$

题解

当$n=1$时，输入只可能是$1$，若$k$为奇数则输出$0$，否则输出1$。

$n>1$时，我们从高位到低位看，把高位的连续的$0$改为$1$是最优的。如果$k$小于连续的$0$的段数，则从高位到低位把连续的$0$改为$1$；如果$k$大于连续的$0$的段数，则将全部的$0$都变为$1$，然后将多余的操作次数消去即可。如果多出偶数次操作，则全部选$[1,1]$则可以互相抵消；如果多出奇数次操作，则可以先消去偶数部分，再用剩下的一次操作消去，比如将操作$[2,5]$分为$[2,3]$和$[4,5]$，或将操作$[1,1]$分为$[1,2]$和$[2,2]$。

还有一种特殊情况，当$k=1$，$s=11\dots 11$时，只能将最低位翻转，得到$11\dots 10$。

时间复杂度为$O(n)$。

code

#include<bits/stdc++.h>
using namespace std;
int t,n,vt;
long long k;
char s[100005];
int main()
{
	scanf("%d",&t);
	while(t--){
		scanf("%d%lld",&n,&k);
		scanf("%s",s+1);
		if(n==1){
			printf("%d\n",(k^1)&1);
			continue;
		}
		vt=0;
		for(int i=1;i<=n;i++){
			if(s[i]=='0'){
				while(i+1<=n&&s[i+1]=='0') ++i;
				++vt;
			}
		}
		if(vt==0&&k==1){
			for(int i=1;i<=n-1;i++) printf("1");
			printf("0\n");
		}
		else{
			for(int i=1;i<=n;i++){
				if(s[i]=='0'){
					if(!k) break;
					--k;
					s[i]='1';
					while(i+1<=n&&s[i+1]=='0') s[++i]='1';
				}
			}
			printf("%s\n",s+1);
		}
	}
	return 0;
}