【暴力+剪枝】输出表演方案 Codeforces Round #545 (Div. 2) B. Circus

Codeforces Round #545 (Div. 2) B. Circus

http://codeforces.com/contest/1138/problem/B

给你两行01串，长度均为n（n<=5000）

第一个串表示第i个人能否扮演小丑（1表示可以，0表示不可以）

第二个串表示第i个人能否扮演杂剧（1表示可以，0表示不可以）

保证n是偶数，让你输出n/2个编号

代表第一次上台

要求第一次上台的可以扮演小丑人数和第二次上台扮演杂剧人数相等，且每个人只能出场一次

直接n^4暴力

加上打剪枝

不会被卡

#include 
using namespace std;
struct node
{
    int x,y,sum;
} a[5005];
char s[5005];
int main()
{
    int n;
    cin>>n;
    int clown=0,play=0,all_2=0,all_0=0;
    scanf("%s",s+1);
    for(int i=1; i<=n; i++)
    {
        a[i].x=s[i]-'0';
    }
    scanf("%s",s+1);
    for(int i=1; i<=n; i++)
    {
        a[i].y=s[i]-'0';
    }

    for(int i=1; i<=n; i++)
    {
        a[i].sum=a[i].x+a[i].y;
        if(a[i].sum==1)
        {
            if(a[i].x) clown++;
            if(a[i].y) play++;
        }
        if(a[i].sum==2) all_2++;
        if(a[i].sum==0) all_0++;
    }

    int h=n/2;
    for(int i=0; i<=clown; i++)
    {
        if(i>h) break;
        if((i+play+all_2+all_0)h) break;
            if((i+j+all_2+all_0)h) break;
                if((i+j+k+all_0)h) break;
                    if(i+j+k+p!=h) continue;
                    if((i+k)!=(play-j+all_2-k)) continue;
                    //cout<