[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"
提示:
-
0 <= A <= 100
-
0 <= B <= 100
-
对于给定的
A
和B
,保证存在满足要求的S
。
思路:当A>B的时候,尽量先输出'a',反之先输出'b';当某一字母连续输出两个时,输出另一个字母。
代码:
class Solution { public String strWithout3a3b(int A, int B) { String ans = ""; int a = 0; int b = 0; while(A>0||B>0) { while(A>0&&((A>=B&&a<2)||b==2)) { ans = ans + 'a'; A--; a++; b=0; } while(B>0&&((B>=A&&b<2)||a==2)) { ans = ans + 'b'; B--; b++; a=0; } } return ans; } }