csu

http://acm.csu.edu.cn/OnlineJudge/problem.php?id=1536

不知道为何怎么写都写不对.

这题可以模拟.

虽然题目保证一定可以从原串变成目标串,但是不一定可以变成两种目标串.

所以需要判断下.统计原串中0和1的个数,然后计算目标串中0可能的个数,1可能的个数.

计算交换次数就是从后面找一个跟当前不一样的数字交换到前面来即可。

#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;
int main()
{
    //freopen("a.txt","r",stdin);
    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++;
    }
    // printf("%d %d
",num1,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];
    }
    //printf("%d %d
",res1,res2);
    if(res1==num1)
    {
        ans=0;
        memset(b,0,sizeof(b));
        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(a[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(a[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
",minn);
    return 0;
}