【CODE】String Without AAA or BBB

984. String Without AAA or BBB

Medium

133200Add to ListShare

Given two integers A and B, return any string S such that:

  • S has length A + B and contains exactly A 'a' letters, and exactly B 'b' letters;
  • The substring 'aaa' does not occur in S;
  • The substring 'bbb' does not occur in S.

Example 1:

Input: A = 1, B = 2
Output: "abb"
Explanation: "abb", "bab" and "bba" are all correct answers.

Example 2:

Input: A = 4, B = 1
Output: "aabaa"

Note:

  1. 0 <= A <= 100
  2. 0 <= B <= 100
  3. It is guaranteed such an S exists for the given A and B.
class Solution {
public:
    string strWithout3a3b(int A, int B) {
        string res="";
        while(B>0 && A>0){
            while(B>0 && A>0 && A>=B){
                if(A-2*B>=0){
                    res+="aab"; A-=2; B-=1;
                }
                else{
                    res+="ab"; A-=1; B-=1;
                }
            }
            while(B>0 && A>0 && B>=A){
                if(B-2*A>=0){
                    res+="bba"; B-=2; A-=1;
                }
                else{
                    res+="ba"; B-=1; A-=1;
                }
            }
        }
        while(B>0){
            res+='b';B--;
        }
        while(A>0){
            res+='a';A--;
        }
        return res;
    }
};
发布了133 篇原创文章 · 获赞 35 · 访问量 6万+

猜你喜欢

转载自blog.csdn.net/Li_Jiaqian/article/details/103885796
bbb