【每日刷题】不含 AAA 或 BBB 的字符串

day31, 不含 AAA 或 BBB 的字符串

题目来源：leetcode
给定两个整数 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”

解答：这道题可以用贪心算法来解决。但是注意需要设置一个flag标志。
我们定义 A, B：待写的 ‘a’ 与 ‘b’ 的数量。
设当前还需要写入字符串的 ‘a’ 与 ‘b’ 中较多的那一个为 x，如果我们已经连续写了两个 x 了，下一次我们应该写另一个字母。否则，我们应该继续写 x。

代码：

string strWithout3a3b(int A, int B) {
        string result = "";
        int flag = 0;
        while( 1){            
            int sum = 0;
            if( A > B && !flag){
                while( sum < 2 && A > 0)
                    result += 'a', A--, sum++;
                if( A > B)
                    flag = 1;
                else
                    flag = 0;
            }
            else if( A > B && flag ){
                if( B > 0)
                    result += 'b', B--,flag = 0;
            }
            else if( A <= B && !flag ){
                while( sum < 2 && B > 0)
                    result += 'b', B--, sum++;
                if( A <= B)
                    flag = 1;
                else
                    flag = 0;
            }
            else if( A <= B && flag){
                if( A > 0)
                    result += 'a', A--, flag = 0;
            }
            if( !A && !B)
                return result;
        }
        return "";
    }

运行结果：