leetcode 984. 不含 AAA 或 BBB 的字符串

目录

• 题目描述：
• 示例 1：
• 示例 2：
• 解法：

题目描述：

给定两个整数 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 {
public:
    string strWithout3a3b(int A, int B) {
        if(A > 2*(B+1) || B > 2*(A+1)){
            return "";
        }else{
            string res = "";
            if(A > B){
                int addition = A - B;
                for(int i = 0; i < B; i++){
                    res += 'a';
                    if(addition > 0){
                        res += 'a';
                        addition--;
                    }
                    res += 'b';
                }
                while(addition--){
                    res += 'a';
                }
            }else{
                int addition = B - A;
                for(int i = 0; i < A; i++){
                    res += 'b';
                    if(addition > 0){
                        res += 'b';
                        addition--;
                    }
                    res += 'a';
                }
                while(addition--){
                    res += 'b';
                }
            }
            return res;
        }
    }
};