csu - 1536: Bit String Reordering (模拟)

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=;
 int a[MAXN],b[MAXN],num[MAXN],temp[MAXN],ans;
 int main()
 {
     //freopen("a.txt","r",stdin);
     int n,m,num0=,num1=,res1=,res2=;
     int minn=0x3f3f3f3f;
     scanf("%d %d",&n,&m);
     for(int i=; i<=n; i++)
     {
         scanf("%d",&a[i]);
         if(a[i]==) num1++;
         if(a[i]==) num0++;
     }
     // printf("%d %d\n",num1,num0);
     for(int i=; i<=m; i++)
     {
         scanf("%d",&temp[i]);
         if(i%==) res1+=temp[i];
         if(i%==) res2+=temp[i];
     }
     //printf("%d %d\n",res1,res2);
     if(res1==num1)
     {
         ans=;
         memset(b,,sizeof(b));
         int flag=,cnt=;
         for(int i=; i<=m; i++)
         {
             for(int j=; j<=temp[i]; j++)
                 b[cnt++]=flag;
             flag=!flag;
         }
         for(int i=; i<=n; i++)
         {
             if(a[i]==b[i]) continue;
             else
             {
                 for(int j=i+; 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=;
         memset(b,,sizeof(b));
         // for(int i=1;i<=n;i++) num[i]=a[i];
         int flag=,cnt=;
         for(int i=; i<=m; i++)
         {
             for(int j=; j<=temp[i]; j++)
                 b[cnt++]=flag;
             flag=!flag;
         }
         for(int i=; i<=n; i++)
         {
             if(a[i]==b[i]) continue;
             else
             {
                 for(int j=i+; 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 ;
 }