CF 1138-B 枚举

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

PS：最近想做一些不太难，但是需要思维，容易错的题。题解怎么快怎么来。

 

给两个n长二进制串，输出n/2个下标，使得选中的下标中，在串1里的“1”的个数==剩下的下标中，在串2里1的个数。

同一个下标下，s1s2的数字看作二元组<0,0><1,0>···，做一个统计存入abcd，可以根据题给条件列出等式，可以得到一个三元一次方程，枚举取得的某两个二元组的个数，就可以推出第三者的个数，验证是否是可行解即可。

可能有点含糊哈，但是没什么时间多写了。

#include <iostream>
#include <algorithm>
#include <string.h>
using namespace std;
typedef long long ll;
const int MAXN = 1e5+8;
const ll mod = 1000000007;
int k,n,ans,cnt;
int kase;
char s1[MAXN],s2[MAXN];
int a,b,c,d;

int main(){
    ios::sync_with_stdio(0);
    cout.tie(0);cin.tie(0);
    cin>>n;
    cin>>(s1+1);
    cin>>(s2+1);

    for(int i=1;i<=n;i++){
        if(s1[i]=='0'&&s2[i]=='0')a++;
        if(s1[i]=='0'&&s2[i]=='1')b++;
        if(s1[i]=='1'&&s2[i]=='0')c++;
        if(s1[i]=='1'&&s2[i]=='1')d++;
    }
    int cc,bb,dd,aa,fd=0;
    for(cc=0;cc<=c;cc++){
        if(cc>n/2)break;
        for(dd=0;dd<=d;dd++){
            if(cc+dd>n/2)break;
            bb=b+d-cc-2*dd;
            aa=n/2-bb-cc-dd;
            if(bb>=0&&bb<=b&&aa>=0&&aa<=a){
                fd=1;
                break;
            }
        }
        if(fd)break;
    }
    if(fd){
        for(int i=1;i<=n;i++){
            if(aa&&s1[i]=='0'&&s2[i]=='0')
                {cout<<i<<" ";aa--;};
            if(bb&&s1[i]=='0'&&s2[i]=='1')
                {cout<<i<<" ";bb--;};
            if(cc&&s1[i]=='1'&&s2[i]=='0')
                {cout<<i<<" ";cc--;};
            if(dd&&s1[i]=='1'&&s2[i]=='1')
                {cout<<i<<" ";dd--;};
        }
        cout<<endl;
    }else{
        cout<<"-1"<<endl;
    }

    return 0;
}
/*
5
*/