acwing 3777.砖块

acwing 3777.砖块

n 个砖块排成一排，从左到右编号依次为 1∼n。

每个砖块要么是黑色的，要么是白色的。

现在你可以进行以下操作若干次（可以是 0 次）：

选择两个相邻的砖块，反转它们的颜色。（黑变白，白变黑）

你的目标是通过不超过 3n 次操作，将所有砖块的颜色变得一致。

输入格式

第一行包含整数 T，表示共有 T 组测试数据。

每组数据第一行包含一个整数 n。

第二行包含一个长度为 n 的字符串 s。其中的每个字符都是 W 或 B，如果第 i个字符是 W，则表示第 i 号砖块是白色的，如果第 i 个字符是 B，则表示第 i 个砖块是黑色的。

输出格式

每组数据，如果无解则输出一行 −1。

否则，首先输出一行 k，表示需要的操作次数。

如果 k>0，则还需再输出一行 k 个整数，p₁,p₂,…,p_k。其中 p_i表示第 i 次操作，选中的砖块为 p_i和 p_i+1 号砖块。

如果方案不唯一，则输出任意合理方案即可。

数据范围

1≤T≤10，
2≤n≤200。

输入样例：

4
8
BWWWWWWB
4
BWBB
5
WWWWW
3
BWB

输出样例：

3
6 2 4
-1
0
2
2 1

#define _CRT_SECURE_NO_WARNINGS 1
/****************************************************/
#include <iostream>
#include <cstring>
#include <algorithm>
#include <vector>
#include<string>
using namespace std;
string s;
int n;
int main()
{
    int t;
    cin >> t;
    while (t--)
    {
        vector<int>w, b;
        cin >> n;
        cin >> s;
        string s0 = s;
        for (int i = 0; i < n - 1; i++)
        {
            if (s[i] == 'W')
            {
                s[i] = 'B';
                b.push_back(i + 1);
                if (s[i + 1] == 'W')s[i + 1] = 'B';
                else s[i + 1] = 'W';
            }
        }
        for (int i = 0; i < n - 1; i++)
        {
            if (s0[i] == 'B')
            {
                s0[i] = 'W';
                w.push_back(i + 1);
                if (s0[i + 1] == 'W')s0[i + 1] = 'B';
                else s0[i + 1] = 'W';
            }
        }
        int a1 = -1, a2 = -1;
        if (s[n - 1] == 'B' && s0[n - 1] == 'B')
        {
            printf("%d\n", b.size());
            for (int i = 0; i < b.size(); i++)
            {
                printf("%d ", b[i]);
            }
            printf("\n");
        }
        else if (s0[n - 1] == 'W' && s[n - 1] == 'W')
        {
            printf("%d\n", w.size());
            for (int i = 0; i < w.size(); i++)
            {
                printf("%d ", w[i]);
            }
            printf("\n");

        }
        else if (s0[n - 1] == 'B' && s[n - 1] == 'W')printf("-1\n");
        else if (b.size() < w.size())
        {
            printf("%d\n", b.size());
            for (int i = 0; i < b.size(); i++)
            {
                printf("%d ", b[i]);
            }
            printf("\n");
        }
        else
        {
            printf("%d\n", w.size());
            for (int i = 0; i < w.size(); i++)
            {
                printf("%d ", w[i]);
            }
            printf("\n");
        }

    }
}