调整队形

2017-04-25  本文已影响0人  RobotBerry

问题描述

在幼儿园有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;
}
上一篇 下一篇

猜你喜欢

热点阅读