不含 AAA 或 BBB 的字符串--先奏后斩思想

0x01.问题

给定两个整数 AB,返回任意字符串 S,要求满足:
1.S 的长度为 A + B,且正好包含 A'a' 字母与 B'b' 字母;
2.子串 'aaa' 没有出现在 S 中;
3.子串 'bbb' 没有出现在 S 中。
提示:0 <= A <= 100 0 <= B <= 100
对于给定的 AB,保证存在满足要求的 S

C++函数形式为:string strWithout3a3b(int A, int B)

0x02.分析

题目保证一定存在这样的S,我们可以逆向的分析一下,当两个整数需要满足什么样的情况时,才一定能保证存在这样的S呢?

来一个极端的思想,如果A非常多,B非常少,AB之间存不存在制约呢?

我们很容易得出这样的极端情况就是A=2B+2,也就是每两个a后都跟着一个b,最后还有两个a的情况。如果a再多一个,就找不到合适的位置了。相反,如果ba少,也是一样的情况。

于是我们就有思路了,假如ab少,我们可以先按照b的数量构造出一个极端情况的字符串,然后,再根据a的数量从两个相邻的里面删去一个。反过来也是一样的做法。

还有就是A==B的情况,这时候,只要一个接一个的加上去就行了。

整个题的思路:

  • 先构造极端条件,再根据实际数量删去相应的重复字符。
  • 先奏后斩,哈哈。

0x03.解决代码

class Solution {
public:
    string strWithout3a3b(int A, int B) {
        string s;
        if(A==B){
            s="";
            for(int i=0;i<A+B;i++){
                if(i%2==0) s+='a';
                else s+='b';
            }
            return s;
        }
        if(A>B){
            s="";
            int a=0,b=0;
            for(int i=0;i<3*B+2;i++){
                if(a<2){
                    s+='a';
                    a++;
                }
                else{
                    s+='b';
                    b++;
                    a=0;
                    if(b>=B){
                        s+="aa";
                        break;
                    }
                }
            }
            int num=2*B+2-A;
            for(int i=0;i<num;i++){
                s.erase(i*2,1);
            }
            return s;
        }
        else{
            s="";
            int b=0,a=0;
            for(int i=0;i<3*A+2;i++){
                if(b<2){
                    s+='b';
                    b++;
                }
                else{
                    s+='a';
                    a++;
                    b=0;
                    if(a>=A){
                        s+="bb";
                        break;
                    }
                }
            }
            int num=2*A+2-B;
            for(int i=0;i<num;i++){
                s.erase(i*2,1);
            }
            return s;
        }
        return s;
    }
};

ATFWUS  --Writing  By 2020–03–20

发布了126 篇原创文章 · 获赞 142 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/ATFWUS/article/details/104988191
bbb