LeetCode 984. 不含 AAA 或 BBB 的字符串(贪心)

1. 题目

给定两个整数 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。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/string-without-aaa-or-bbb
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

2. 贪心

  • 先用A,B的较小的数 n = m i n A , B n=min(A,B) 生成 n 个 ab,或者 ba, 谁多,谁打头
  • 然后剩余的 a 或者 b 插空,即可
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;
    }
};

在这里插入图片描述

发布了587 篇原创文章 · 获赞 454 · 访问量 7万+

猜你喜欢

转载自blog.csdn.net/qq_21201267/article/details/103883367