csu 1536 Bit String Reordering(模拟)

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<stdlib.h>
#include<algorithm>
using namespace std;
const int MAXN=20;
int a[MAXN],b[MAXN],num[MAXN],temp[MAXN],ans;
void swap(int &a,int &b)
{
    int temp;
    temp=a;
    a=b;
    b=temp;
}
int min(int a,int b)
{
    if(a>b) return b;
    else return a;
}
int main()
{
    int n,m,num0=0,num1=0,res1=0,res2=0;
    int minn=0x3f3f3f3f;
    scanf("%d %d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        if(a[i]==1) num1++;
        if(a[i]==0) num0++;
    }
    for(int i=1;i<=m;i++)
    {
        scanf("%d",&temp[i]);
        if(i%2==1) res1+=temp[i];
        if(i%2==0) res2+=temp[i];
    }
    if(res1==num1)
    {
        ans=0;
        memset(b,0,sizeof(b));
        for(int i=1;i<=n;i++) num[i]=a[i];
        int flag=1,cnt=1;
        for(int i=1;i<=m;i++)
        {
            for(int j=1;j<=temp[i];j++)
                b[cnt++]=flag;
            flag=!flag;
        }
        for(int i=1;i<=n;i++)
        {
            if(num[i]==b[i]) continue;
            else
            {
                for(int j=i+1;j<=n;j++)
                {
                    if(b[j]==!b[i])
                    {
                        swap(b[j],b[i]);
                        ans+=j-i;
                        break;
                    }
                }
            }
        }
        minn=min(ans,minn);
    }
    if(res1==num0)
    {
        ans=0;
        memset(b,0,sizeof(b));
        for(int i=1;i<=n;i++) num[i]=a[i];
        int flag=0,cnt=1;
        for(int i=1;i<=m;i++)
        {
            for(int j=1;j<=temp[i];j++)
                b[cnt++]=flag;
            flag=!flag;
        }
        for(int i=1;i<=n;i++)
        {
            if(num[i]==b[i]) continue;
            else
            {
                for(int j=i+1;j<=n;j++)
                {
                    if(b[j]==!b[i])
                    {
                        swap(b[j],b[i]);
                        ans+=j-i;
                        break;
                    }
                }
            }
        }
        minn=min(ans,minn);
    }
    printf("%d\n",minn);
    return 0;
}