问题描述
在幼儿园有n个小朋友排列为一个队伍,从左到右一个挨着一个编号为(0~n-1)。其中有一些是男生,有一些是女生,男生用'B'表示,女生用'G'表示。小朋友们都很顽皮,当一个男生挨着的是女生的时候就会发生矛盾。作为幼儿园的老师,你需要让男生挨着女生或者女生挨着男生的情况最少。你只能在原队形上进行调整,每次调整只能让相邻的两个小朋友交换位置,现在需要尽快完成队伍调整,你需要计算出最少需要调整多少次可以让上述情况最少。例如:
GGBBG -> GGBGB -> GGGBB
这样就使之前的两处男女相邻变为一处相邻,需要调整队形2次
输入描述
输入数据包括一个长度为n且只包含G和B的字符串.n不超过50.
输出描述
输出一个整数,表示最少需要的调整队伍的次数
输入例子
GGBBG
输出例子
2
分析
初看起来和逆序对的问题很像,逆序对可以在各种排序算法中进行统计。但是题目要求男生和女生只能和相邻的小朋友交换位置,不能和队伍中的任何一个人交换位置。于是想到two-pointers的方法,首尾指针遍历,遇到顺序不对的就进行交换,累加计数器。注意最终的结果可能是GGG...BBB...,也可能是BBB...GGG...,两种情况都需要考虑,最后取交换次数较小的那个。
note
不需要真正交换元素,遍历到顺序不对的元素只累加计数器就好了
代码
#include <cstdio>
#include <cstring>
#include <vector>
#include <algorithm>
using namespace std;
// 计算交换次数,bc(begin character)为队首性别,ec(end character)为队尾性别
int calcSwapCount(char bc, char ec, char *str, int n)
{
int count = 0;
int begin = 0, end = n - 1;
while (begin < end)
{
if (str[begin] == bc)
begin++;
else if (str[end] == ec)
end--;
else if (str[begin] == ec && str[end] == bc)
{
count += end - begin; // 将顺序出错的begin和end交换需要的步数
begin++;
end--;
}
}
return count;
}
int main()
{
char str[51];
scanf("%s", &str);
int n = strlen(str);
printf("%d\n", min(calcSwapCount('G', 'B', str, n),
calcSwapCount('B', 'G', str, n)));
return 0;
}