调整队形

xiaoxiao2021-02-28  92

问题 在幼儿园有 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)); } }
转载请注明原文地址: https://www.6miu.com/read-31710.html

最新回复(0)