[数据结构与算法] B. Bit Flipping

开发: C++知识库 Java知识库 JavaScript Python PHP知识库人工智能区块链大数据移动开发嵌入式开发工具数据结构与算法开发测试游戏开发网络协议系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑笔记本显卡显示器固态硬盘硬盘耳机手机 iphone vivo oppo 小米华为单反装机图拉丁

-> 数据结构与算法 -> B. Bit Flipping -> 正文阅读

[数据结构与算法]B. Bit Flipping

#include<iostream>
#include<string>
using namespace std;
int main(){
	int t=0;
	cin>>t;
	while(t--){
		int n=0,k=0;
		string a;
		int b[200005]={0};
		cin>>n>>k;//k奇动1，k偶动0
			cin>>a;
            //以上为输入
		if(k%2){//k奇动1
			int j=0;
			for(int i=0;i<n;i++){
				if(a[i]=='1'&&j<k){
					b[i]++;
					j++;
				}	
			}
			b[n-1]+=k-j;
			for(int i=0;i<n;i++){
				if(b[i]%2) cout<<a[i];
				else cout<<!(a[i]-'0');
			}
			cout<<endl;
			for(int i=0;i<n;i++){
				cout<<b[i]<<' ';
			}
			cout<<endl;
		}else{//k偶动0
			int j=0;
			for(int i=0;i<n;i++){
				if(a[i]=='0'&&j<k){
					b[i]++;
					j++;
				}	
			}
			b[n-1]+=k-j;
			for(int i=0;i<n;i++){
				if(b[i]%2) cout<<!(a[i]-'0');
				else cout<<a[i];
			}
			cout<<endl;
			for(int i=0;i<n;i++){
				cout<<b[i]<<' ';
			}
			cout<<endl;
		}
	}
	return 0;
}

啧，真长，优化一下，把奇偶合并

#include<iostream>
#include<string>
using namespace std;
int main(){
	int t=0;
	cin>>t;
	while(t--){
		int n=0,k=0;
		string a;
		int b[200005]={0};
		cin>>n>>k;//k奇动1，k偶动20
		cin>>a;
		int j=0,p=k%2;
		for(int i=0;i<n;i++){
			if(a[i]-'0'==p&&j<k){
				b[i]++;
				j++;
			}	
		}
		b[n-1]+=k-j;
		for(int i=0;i<n;i++){
			if(b[i]%2==p) cout<<a[i];
			else cout<<!(a[i]-'0');	
		}
		cout<<endl;
		for(int i=0;i<n;i++){
			cout<<b[i]<<' ';
		}
		cout<<endl;
	}
	return 0;
}