B. Bit Flipping
阿新 • • 發佈:2022-04-18
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; }