Python实现字符串去除驼峰

题目

字符串去除驼峰，字符串驼峰指相邻的三个字符两端的两个字符相同，如：aba、AcA
式例：
输入：abcbef，输出：aef
输入：AbAcef，输出：cef

实现

def solution(array: str) -> str:
    array_list = []
    i = 0
    while i < len(array) - 2:
        if array[i] == array[i + 2]:
            i += 3
        else:
            array_list.append(array[i])
            i += 1
    array_list.extend(array[-2:])
    return ''.join(array_list)


if __name__ == '__main__':
    print(solution('abcbef'))
    print(solution('AbAcef'))