给定两个整数 A 和 B,返回任意字符串 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。
来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/string-without-aaa-or-bbb 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
生成 n 个 ab
,或者 ba
, 谁多,谁打头
class Solution {
public:
string strWithout3a3b(int A, int B) {
int n = min(A,B), i = 0;
A -= n;
B -= n;
string ans;
while(n--)
if(A>=B)
ans += "ab";
else
ans += "ba";
if(A)
{
while(A--)
{
if(i > ans.size())
ans.push_back('a');
else
ans.insert(i,1,'a');
i += 3;
}
}
if(B)
{
while(B--)
{
if(i > ans.size())
ans.push_back('b');
else
ans.insert(i,1,'b');
i += 3;
}
}
return ans;
}
};