B. Circus

链接

[https://codeforces.com/problemset/problem/1138/B]

题意

给你两个字符串0，1那种，你是否可以选出一半的字符串，同时取。
使得当前第一个字符串的1的个数等于，剩下没选的第二个字符串的1的个数。

分析

首先我是真的傻，因为，我一直在哪里分类讨论，结果想了半天，一点都不行。
那么这个时候我们就看数据n<=5000,也就是可以n*n解决。怎么暴力呢？
这题就是你初中的数学功底的体现了。统计00，01，10，11的个数A,B,C,D；
我们假设选的00，01，10，11的个数为a,b,c,d;
a+b+c+d=n/2; A+B+C+D=n;
我们枚举a,那么根据题意需要满足

这题真的就是个数学啊，而且也反应我对复杂度不敏感，有时候可以暴力，就一定暴力，因为它不容易出错比较稳。
复杂度分析，这方面我还是菜了。

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=5e3+10;

char s[N],t[N];
int main(){
	int n;
	while(~scanf("%d",&n)){
	
		scanf("%s",s+1);
		scanf("%s",t+1);
		int a,b,c,d,A=0,B=0,D=0,C=0;
		for(int i=1;i<=n;i++)
		if(s[i]=='0'&&t[i]=='1')
		B++;
		else if(s[i]=='1'&&t[i]=='0')
		C++;
		else if(s[i]=='1'&&t[i]=='1')
		D++;
		A=n-(B+D+C);
		//cout<<A<<' '<<B<<' '<<C<<' '<<D<<endl;
		bool flag=0;
		for(a=0;a<=A&&a<=n/2;a++)
		{
			d=B+D+a-n/2;
			if(d>=0&&d<=D&&(a+d)<=n/2){
				for(b=0;b<=B&&(a+b+d)<=n/2;b++){
					c=n/2-(a+b+d);
					if(c>=0&&c<=C){
						flag=1; break;
					}
				}
			}
			if(flag) break;
		}
		if(!flag) puts("-1");
		else {
			//cout<<a<<' '<<b<<' '<<c<<' '<<d<<endl;
			for(int i=1;i<=n;i++)
			if(s[i]=='0'){
				if(t[i]=='0'&&a){
					printf("%d ",i);
					a--;
				}
				if(t[i]=='1'&&b){
					printf("%d ",i);
					b--;
				}
			}
			else{
				if(t[i]=='0'&&c){
					printf("%d ",i);
					c--;
				}
				if(t[i]=='1'&&d){
					printf("%d ",i);
					d--;
				}
			}
			puts("");
		}
	}
	return 0;
}