Leetcode-984： 不含 AAA 或 BBB 的字符串

题目链接

https://leetcode-cn.com/problems/string-without-aaa-or-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"

提示

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

思路

这题学习了MskAdr的思想。

1. 当a和b其中一个为0，直接将另一个全部输出即可。

2. 当a和b相等，也直接拼接成ababab...形式即可。

3. 当a和b不相等，且a和b都不为0，也就是除了上述两种情况剩余的情况，a和b之间的个数差最大也就为3，也就是需要排列成aabaa，bbabb这种极端情况，如果个数相差4以上，显然无解，因此我们就按照最差的情况来，如果a>b那么就先把aab排好，如果b>a，就先把bba排好，先最大化的排多的那个字符，再把剩下的字符按照前面的思路继续排列（递归）。

C++ Code

class Solution {
public:
    string strWithout3a3b(int a, int b) {
        if(b==0) return string(a,'a');
        if(a==0) return string(b,'b');
        if(a==b) return "ab"+strWithout3a3b(a-1,b-1);
        return a>b?"aab"+strWithout3a3b(a-2,b-1):"bba"+strWithout3a3b(a-1,b-2);
    }
};

结果