Round545div2B(1138B)

一、题目链接

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

二、思路

贪心是肯定不行的。

设会$[1,0]$的人存在容器$p_1$里面,会$[0,1]$的人存在容器$p_2$里面,会$[1,1]$的人存在容器$p_3$里面,会$[0,0]$的人存在容器$p_4$里面。

设$p_1.size()==s_1$,$p_2.size()==s_2$,$p_3.size()==s_3$,$p_4.size()==s_4$。

枚举在$p_1$部分选择的人数$i$,在$p_3$部分选择的人数$j$,那么,$p_1$部分剩余的$s_1-i$个人和$p_3$部分剩余的$s_3-j$个人,一定要被分到第二场演出(第二部分)。这时,把所有非法情况判掉,答案就出来了。

非法情况如下:

1.$s_2+s_3-j<i+j$,剩余的第二部分的人数不够枚举出来的第一部分的人数;

2.$s_2<s_2+s_3-j-(i+j)$,为了和第一部分枚举的人数平衡,第二部分多出来的属于$p_2$容器里面的人,都必须要放到第一部分去,如果把所有$p_2$容器里面的人全部放到第一部分后,剩余的第二部分的人(全部来自$p_3$,数量就是$s_3-j$),还是比$i+j$多,那么,也就是满足当前这个条件,非法;

为表示方便,用$part2sub=s_2+s_3-j-(i+j)$,表示属于容器$p_2$的人要被放到第一部分的数量。

3.$i+j+part2sub>\frac{n}{2}$,枚举的第一部分的人数,加上容器$p_2$里面被强制放过来的人数,多于一半,非法;

4.$s_2+s_3-j-part2sub+s_1-i>\frac{n}{2}$,$s_2+s_3-j-part2sub$是为了和枚举的第一部分的人数相同而计算出来的第二部分的且在第二部分起作用的人数(起作用就是说,它是$p_2$或者$p_3$容器里面的人),$s_1-i$是属于容器$p_1$的人被强制放到第二部分的数量,这两项加起来,就是为了排除上述非法条件而强制放到第二部分的人数。如果这两项之和大于总数的一半,非法;

排除了上述4个条件,答案就出来了。在容器$p_1$取$i$个人,在容器$p_3$取$j$个人,在容器$p_2$取$part2sub$个人,剩余的$\frac{n}{2}-i-j-part2sub$个人,去$p_4$里面取。

三、代码

//
// Created by fuzhihong on 3/8/19.
//

#include<bits/stdc++.h>

using namespace std;
int n, vis[5010];
char a[5010], b[5010];
vector<int> res, p1, p2, p3, p4, p5;

int main() {
    cin >> n;
    scanf("%s", a + 1);
    scanf("%s", b + 1);
    for (int i = 1; i <= n; ++i) {
        if (a[i] == '1' && b[i] == '0')p1.push_back(i);
        if (a[i] == '0' && b[i] == '1')p2.push_back(i);
        if (a[i] == '1' && b[i] == '1')p3.push_back(i);
        if (a[i] == '0' && b[i] == '0')p4.push_back(i);
    }
    int s1 = p1.size(), s2 = p2.size(), s3 = p3.size(), s4 = p4.size();
    for (int i = 0; i <= s1; ++i) {
        for (int j = 0; j <= s3; ++j) {
            int part2 = s2 + s3 - j;
            if (part2 < i + j)continue;
            if (s2 < part2 - (i + j))continue;
            int part2sub = part2 - (i + j);
            if (i + j + part2sub > n / 2)continue;
            if (part2 - part2sub + s1 - i > n / 2)continue;
            for (int k = 0; k < i; ++k)printf("%d ", p1[k]);
            for (int k = 0; k < j; ++k)printf("%d ", p3[k]);
            for (int k = 0; k < part2sub; ++k)printf("%d ", p2[k]);
            for (int k = i + j + part2sub, x = 0; k < n / 2; ++k, ++x)printf("%d ", p4[x]);
            //printf("\n%d %d\n",i,j);
            return 0;
        }
    }
    puts("-1");
    return 0;
}

/*
 *
 4
 0011
 1100

 4
 0011
 0011

  4
 0011
 0000

 8
 01100000
 10100000

 10
 1111100000
 0000001111


 *
 * **/

 

posted @ 2019-03-10 09:17 fuzhihong0917 阅读(...) 评论(...) 编辑 收藏