调整队形

问题

在幼儿园有 n 个小朋友排列为一个队伍，从左到右一个挨着一个编号为 ( 0 ~ n-1 ) 。其中有一些是男生，有一些是女生，男生用 ’B’ 表示，女生用 ’G’ 表示。小朋友们都很顽皮，当一个男生挨着的是女生的时候就会发生矛盾。作为幼儿园的老师，你需要让男生挨着女生或者女生挨着男生的情况最少。你只能在原队形上进行调整，每次调整只能让相邻的两个小朋友交换位置，现在需要尽快完成队伍调整，你需要计算出最少需要调整多少次可以让上述情况最少。例如：

GGBBG -GGBGB -GGGBB 
这样就使之前的两处男女相邻变为一处相邻，需要调整队形 2 次

输入描述

输入数据包括一个长度为 n 且只包含 G 和 B 的字符串. n 不超过 50.

输出描述

输出一个整数，表示最少需要的调整队伍的次数

例子

输入：GGBBG
输出：2

解题思路

从最后会形成的结果入手，我们可以得知，所有的男孩与所有的女孩各自在一边，只有中间的一男一女会紧挨着，如'GGBBG'要么变成'BBGGG'或'GGGBB'，只用比较这两个方向调整的方法所花的步骤谁少，即为最终结果。

代码

import java.util.ArrayList;
import java.util.Scanner;

public class Test {

    public static void main(String[] args) {
        // 控制台输入
        Scanner sc = new Scanner(System.in);
        String s = sc.next();
        sc.close();

        int n = s.length();
        if (n <= 0 || n > 50) {
            throw new RuntimeException("the length is too long");
        }
        ArrayList<Integer> list = new ArrayList<>();
        for (int i = 0; i < n; i++) {
            if (s.charAt(i) == 'B') {
                list.add(i);
            }
        }

        int bSize = list.size();
        int indexSum = 0;
        for (int val : list) {
            indexSum += val;
        }
        //最关键的部分，以left的求解为例，indexSum指每个元素移到最左边所花的步骤
        //然而实际最左边只有一个位置，多余的元素都需要依次往右边排开，可以看作再往右边依次移动
        int left = indexSum - bSize * (bSize - 1) / 2;
        int right = (bSize * (n - 1) - indexSum) - bSize * (bSize - 1) / 2;
        // 移左 或者 移右 ，选择最少的
        System.out.println(Math.min(left, right));
    }
}