问题
在幼儿园有 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));
}
}