【B. Blocks】 CodeForces - 1271B

B.Blocks

题意：

就是给你一个字符串，例如：BWWWWWWB，W 表示白色， B 表示蓝色，然后经过几次操作之后，可以让这些变为同一颜色，所不能的话，输出 -1 ，若能的话，先输出操作的次数，接着输出所操作的字符下标。

只能将相邻位置和当前位置进行变色，如果是 WB 可以变成 BW，如果是 BB 话，可以变成 WW 。

思路：

有三种情况：

• 当白色和蓝色的数量都是奇数时，无论怎么操作，都不能变成同一颜色，输出 - 1；
• 当一种颜色的数量为 0 的时候，表示不需要操作；
• 当这两种颜色的数量一个为偶数一个为奇数的时候，则一定能变成同一颜色；

做法：

• 当为第三种情况的时候，我们需要找数量为偶数的颜色，并且将这种颜色全部变成另一种颜色，假如 B 的颜色为偶数，则会将所有的 B 变成 W 。（若两种颜色的数量都为偶数的情况下，任意选择一种颜色进行操作即可）

• 当前位置 i 的颜色假如为 B （假如 B 颜色的数量为偶数），当位置为 i + 1 的颜色为 B 时，则将这两个位置的颜色全部变成 W ，并将 i 移动到操作完之后的位置。

• 当 i + 1 位置的颜色为 W 时，则交换这两个位置的颜色；

还有一点，就是当我们选择要操作的颜色的时候，碰到我们选择颜色时，我们就要将此位置保存下来，因为我们一定会操作此位置！

AC code：

#include <map>
#include <queue>
#include <stack>
#include <math.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include <algorithm>

using namespace std;

const int N = 1010,INF=0x3f3f3f3f;
typedef long long ll;

// 升序
//priority_queue <int,vector<int>,greater<int> > q;
// 降序
//priority_queue <int,vector<int>,less<int> >q;

char arr[N];
int b[N];


void swap(char *a,char *b)
{
    char t=*a;
    *a=*b;
    *b=t;
}
int main()
{
    int n,lenw=0,lenb=0,k=0;
    scanf("%d%*c",&n);
    scanf("%s",arr+1);

    for(int i=1;i<=n;i++)
    {
        if(arr[i]=='W')
            lenw++;
        else
            lenb++;
    }
    if(lenb%2&&lenw%2)  
    	printf("-1\n");
    else if((!lenw&&lenb)||(!lenb&&lenw)) 
    	printf("0\n");
    else
    {
        if(!(lenb%2))
        {
            for(int i=1;i<=n-1;i++)
            {
                if(arr[i]=='B')
                {
                    b[++k]=i;
                    if(arr[i+1]=='B')
                    {
                        arr[i]=arr[i+1]='W';
                        i++;
                    }
                    else
                        swap(&arr[i],&arr[i+1]);
                }
            }
        }
        else
        {
            if(!(lenw%2))
            {
                for(int i=1;i<=n-1;i++)
                {
                    if(arr[i]=='W')
                    {
                        b[++k]=i;
                        if(arr[i+1]=='W')
                        {
                            arr[i]=arr[i+1]='B';
                            i++;
                        }
                        else
                            swap(&arr[i],&arr[i+1]);
                    }
                }
            }
        }
         printf("%d\n",k);
        for(int i=1;i<=k-1;i++)
        {
            printf("%d ",b[i]);
        }
        printf("%d\n",b[k]);
    }
    return 0;
}