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"
解释:这是该测试用例的唯一正确答案。

思路:每次从a,b,c中个数最多的字母tmp,该tmp(比如a)能否放入结果res,分三种情况:

1、tmp能放入res,则放入(res倒数第1、第2个字母不同时为tmp)
2、tmp不能放入res,并且没有其他字符可选(比如b、c个数为0),则结束
3、tmp不能放入res,选择个数次多的元素放入(比如b,一定能放入)

class Solution:
    def longestDiverseString(self, a: int, b: int, c: int) -> str:
        m = {'a':a, 'b':b, 'c':c}
        re = ''
        while True:
            #re中每增加一个元素,都要对剩下的a,b,c个数进行排序!!!
            tmp = sorted([(k, v) for k, v in m.items()], key = lambda x:x[1], reverse = True)
            print(re)
            if len(re)>=2 and re[-1] == re[-2] and re[-1] == tmp[0][0]:
                if m[tmp[1][0]] <=0:
                    break
                re += tmp[1][0]
                m[tmp[1][0]] -= 1
            else:
                if m[tmp[0][0]] <= 0:
                    break
                re += tmp[0][0]
                m[tmp[0][0]] -= 1
        return re

 

posted @ 2020-05-23 10:23  USTC丶ZCC  阅读(238)  评论(0编辑  收藏  举报