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

最新推荐文章于 2024-09-17 14:09:50 发布

原创最新推荐文章于 2024-09-17 14:09:50 发布 · 397 阅读

0 ·

CC 4.0 BY-SA版权

暴力大法好专栏收录该内容

4 篇文章

订阅专栏

本文深入解析了Codeforces Round#545(Div.2)B题“Circus”的解决策略，通过n^4暴力算法加剪枝优化，实现了在特定条件下找到满足小丑与杂剧表演者数量相等的人员分配方案。文章详细展示了算法实现过程，包括输入处理、条件判断及输出结果。

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 <bits/stdc++.h>
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) continue;
        for(int j=0; j<=play; j++)
        {
            if((i+j)>h) break;
            if((i+j+all_2+all_0)<h) continue;
            for(int k=0; k<=all_2; k++)
            {
                if((i+j+k)>h) break;
                if((i+j+k+all_0)<h) continue;
                for(int p=0; p<=all_0; p++)
                {
                    if((i+j+k+p)>h) break;
                    if(i+j+k+p!=h) continue;
                    if((i+k)!=(play-j+all_2-k)) continue;
                    //cout<<i<<" "<<j<<" "<<k<<" "<<p<<endl;
                    int a1=0,b1=0,c1=0,d1=0;
                    for(int o=1; o<=n; o++)
                    {
                        if(a[o].x==1&&a[o].y==0&&a1!=i)
                        {
                            a1++;
                            printf("%d ",o);
                        }
                    }
                    for(int o=1; o<=n; o++)
                    {
                        if(a[o].x==0&&a[o].y==1&&b1!=j)
                        {
                            b1++;
                            printf("%d ",o);
                        }
                    }
                    for(int o=1; o<=n; o++)
                    {
                        if(a[o].sum==2&&c1!=k)
                        {
                            c1++;
                            printf("%d ",o);
                        }
                    }
                    for(int o=1; o<=n; o++)
                    {
                        if(a[o].sum==0&&d1!=p)
                        {
                            d1++;
                            printf("%d ",o);
                        }
                    }
                    cout<<endl;
                    return 0;
                }
            }
        }
    }
    printf("-1\n");
    return 0;
}