1. 程式人生 > 其它 >B. Bit Flipping

B. Bit Flipping

B. Bit Flipping

思路:分情況討論,一個數\(x\)翻轉偶數次不變,翻轉奇數次變成x^1

1.\(k\)為奇數,則如果一個數不被選到,那麼翻轉k(奇數次),則原來是\(0\)就能變成1,原來是\(1\)就變成\(0\)。那麼,我們肯定優先從左到右,優先翻轉原來為\(1\)的,這樣前面儘可能多的的\(1\)被翻轉了\(k - 1\)次,翻轉偶數次還是1。

2.k為偶數,同理,我們肯定優先從左到右優先翻轉原來為\(0\)的。

3.假設操作完\(1\)或者2以後還有剩餘,就把剩餘的\(k\)全部給最後一個。假如最後剩下的\(k\)為偶數,則翻轉沒有任何效果。假如最後剩下的\(k\)為奇數,為了使字典序最大,優先把最後一個改成\(0\)
(因為k多出來了,前面肯定都是\(1\)了)。

程式碼:

const int N = 200010;
int n, k;
char str[N];
int ans[N];

void solve()
{
	cin >> n >> k;
	scanf("%s", str + 1);
	for(int i = 1; i <= n; i ++ ) ans[i] = 0;
	if(k % 2 == 1){
		for(int i = 1; i <= n; i ++ ){
			if(str[i] == '1' && k){
				k --;
				ans[i] ++;
			}else if(str[i] == '1'){
				str[i] = '0';
			}else{
				str[i] = '1';
			}
		}
		ans[n] += k;
		if(k % 2 == 1){
			if(str[n] == '1') str[n] = '0';
			else str[n] = '1';
		}
	}else{
		for(int i = 1; i <= n; i ++ ){
			if(str[i] == '0' && k){
				k --;
				str[i] = '1';
				ans[i] ++;
			}
		}
		ans[n] += k;
		if(k % 2 == 1){
			if(str[n] == '1') str[n] = '0';
			else str[n] = '1';
		}		
	}
	printf("%s\n", str + 1);
	for(int i = 1; i <= n; i ++ ){
		cout << ans[i] << " ";
	}
	cout << endl;
}