Leetcode-Easy 984. String Withou

题目描述

给定两个整数A和B，A代表‘a’的个数，B代表‘b’的个数，字符串的长度为A+B，输出一个字符串，字符串中不能出现‘aaa’或者‘bbb’

例1：

Input: A = 1, B = 2
Output: "abb"
Explanation: "abb", "bab" and "bba" are all correct answers.

例2：

Example 2:

Input: A = 4, B = 1
Output: "aabaa"

思路

字符串a和b组合情况其实与A，B大小有关。如果A>=B,字符串末尾添加‘a’然后A-1。如果A<B或者字符串末尾最后两个字符连续出现相同的两个‘a’，字符串末尾添加‘b’，然后B-1

代码实现

class Solution(object):
    def strWithout3a3b(self, A, B):
        ans = []

        while A or B:
            if len(ans) >= 2 and ans[-1] == ans[-2]:
                writeA = ans[-1] == 'b'
            else:
                writeA = A >= B

            if writeA:
                A -= 1
                ans.append('a')
            else:
                B -= 1
                ans.append('b')

        return "".join(ans)