leetcode_984 不含 AAA 或 BBB 的字符串

要求

给定两个整数 A 和 B，返回任意字符串 S，要求满足：

S 的长度为 A + B，且正好包含 A 个 'a' 字母与 B 个 'b' 字母；
子串 'aaa' 没有出现在 S 中；
子串 'bbb' 没有出现在 S 中。  

示例 1：

输入：A = 1, B = 2
输出："abb"
解释："abb", "bab" 和 "bba" 都是正确答案。

示例 2：

输入：A = 4, B = 1
输出："aabaa"

提示：

• 0 <= A <= 100
• 0 <= B <= 100
• 对于给定的 A 和 B，保证存在满足要求的 S。

核心代码

class Solution:
    def strWithout3a3b(self, a: int, b: int) -> str:
        ans = []
        while a or b:
            if len(ans) >= 2 and ans[-1] == ans[-2]:
                write_a = ans[-1] == "b"
            else:
                write_a = a >= b
            if write_a:
                a -= 1
                ans.append("a")
            else:
                b -= 1
                ans.append("b")
        return "".join(ans)

解题思路：我们每次写一个字母，我们都判读这个字符写了几次了，如果写了两次了，我们就进行一次反转，在写另一个字符，所以最核心的就是每次在进行写前都要进行判断。