Prefix Flip (Easy Version)/(Hard Version)

题目

思路

• 做法：先把a全部变为1，或者变为0，做法是从前完后遍历，如果an和an+1不相同，就把前an做一次反转，这样就和an+1相同了，这样最多做n次，然后再从后往前遍历b和a，如果bi≠ai，就把ai前缀反转即可，这样最多n次，所以总共2n次

代码

#include<stdlib.h>
#include<iostream>
#include<stdio.h>
#include<string.h>
using namespace std;
char a[100005],b[100005];
int ans[200009];
int main(){
	int t;
	cin>>t;
	while(t--){
		int cnt=0;
		int n;
		cin>>n;
		for(int j=1;j<=n;j++){
			cin>>a[j];
		}
		for(int j=1;j<=n;j++){
			cin>>b[j];
		}
		for(int j=1;j<=n-1;j++){
			if(a[j]==a[j+1]){continue;}
			if(a[j]!=a[j+1]){ans[cnt++]=j;}
		}
		char c=a[n];
		for(int j=n;j>=1;j--){
			if(b[j]==c)continue;
			if(b[j]!=c){
			    if(c=='1'){
			        c='0';
			    }else{
			        c='1';
			    }
			
			ans[cnt++]=j;}
		}
		cout<<cnt<<" ";
		for(int k=0;k<cnt;k++){cout<<ans[k]<<" ";}
		cout << "\n";
	}
	return 0;
}