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”

提示

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

解题思路

贪心算法：
每次在写入字符串的时候，应该选择当前所剩最多的待写字母x，只有前两个已经写入的字母都是x时候才不选择它

代码

/**
 * @Author: Dunfu Peng
 * @Date: 2019/4/13 12:45
 */
public class Solution {

    public String strWithout3a3b(int A, int B) {
        StringBuilder ans = new StringBuilder();

        while (A > 0 || B > 0) {
            boolean writeA = false;
            int L = ans.length();
            if (L >= 2 && ans.charAt(L - 1) == ans.charAt(L - 2)) {
                if (ans.charAt(L - 1) == 'b') {
                    writeA = true;
                }
            } else {
                if (A >= B) {
                    writeA = true;
                }
            }

            if (writeA) {
                A--;
                ans.append('a');
            } else {
                B--;
                ans.append('b');
            }
        }
        return ans.toString();
    }
}

---

欢迎关注我的微信公众号，接收最新动态。