快乐字符串

Posted 2022-02-18 Roam-G

1405. 最长快乐字符串

难度中等

如果字符串中不含有任何 'aaa'，'bbb' 或 'ccc' 这样的字符串作为子串，那么该字符串就是一个「快乐字符串」。

给你三个整数 a，b ，c，请你返回 任意一个 满足下列全部条件的字符串 s：

• s 是一个尽可能长的快乐字符串。

• s 中 最多 有a 个字母 'a'、b 个字母 'b'、c 个字母 'c' 。

• s 中只含有 'a'、'b' 、'c' 三种字母。

如果不存在这样的字符串 s ，请返回一个空字符串 ""。

示例 1：

输入：a = 1, b = 1, c = 7 输出："ccaccbcc" 解释："ccbccacc" 也是一种正确答案。

示例 2：

输入：a = 2, b = 2, c = 1 输出："aabbc"

示例 3：

输入：a = 7, b = 1, c = 0 输出："aabaa" 解释：这是该测试用例的唯一正确答案。

提示：

• 0 <= a, b, c <= 100
• a + b + c > 0

-*******

方法一：贪心

思路

题目要求找到最长的快乐字符串，且快乐字符串中不含有三个连续相同的字母。为了找到最长的字符串，我们可以使用如下贪心策略：

尽可能优先使用当前数量最多的字母，因为最后同一种字母剩余的越多，越容易出现字母连续相同的情况。如果构建完成最长的快乐字符串后还存在剩余未选择的字母，则剩余的字母一定为同一种字母且该字母的总数量最多。

依次从当前数量最多的字母开始尝试，如果发现加入当前字母会导致出现三个连续相同字母，则跳过当前字母，直到我们找到可以添加的字母为止。实际上每次只会在数量最多和次多的字母中选择一个。

如果尝试所有的字母都无法添加，则直接退出，此时构成的字符串即为最长的快乐字符串。

Java

class Solution 
    public String longestDiverseString(int a, int b, int c) 
        StringBuilder res = new StringBuilder();
        Pair[] arr = new Pair(a, 'a'), new Pair(b, 'b'), new Pair(c, 'c');
        
        while (true) 
            Arrays.sort(arr, (p1, p2) -> p2.freq - p1.freq);
            boolean hasNext = false;
            for (Pair pair : arr) 
                if (pair.freq <= 0) 
                    break;
                
                int m = res.length();
                if (m >= 2 && res.charAt(m - 2) == pair.ch && res.charAt(m - 1) == pair.ch) 
                    continue;
                
                hasNext = true;
                res.append(pair.ch);
                pair.freq--;
                break;
            
            if (!hasNext) 
                break;
            
        
      
        return res.toString();
    

    class Pair 
        int freq;
        char ch;

        public Pair(int freq, char ch) 
            this.freq = freq;
            this.ch = ch;
        
    

python

class Solution:
    def longestDiverseString(self, a: int, b: int, c: int) -> str:
        ans = []
        cnt = [[a, 'a'], [b, 'b'], [c, 'c']]
        while True:
            cnt.sort(key=lambda x: -x[0])
            hasNext = False
            for i, (c, ch) in enumerate(cnt):
                if c <= 0:
                    break
                if len(ans) >= 2 and ans[-2] == ch and ans[-1] == ch:
                    continue
                hasNext = True
                ans.append(ch)
                cnt[i][0] -= 1
                break
            if not hasNext:
                return ''.join(ans)