leetcode [984] 不含 AAA 或 BBB 的字符串

---

Contact me:

Blog -> https://cugtyt.github.io/blog/index
Email -> cugtyt@qq.com
GitHub -> Cugtyt@GitHub

---

给定两个整数 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:
        res = ''
        while A and B:
            if A > B:
                res += 'aab'
                A, B = A - 2, B - 1
            elif A < B:
                res += 'bba'
                A, B = A - 1, B - 2
            else:
                res += 'ab'
                A, B = A - 1, B - 1
        if A: res += 'a' * A
        if B: res += 'b' * B
        return res