leetcode-最长快乐字符串

182 阅读2分钟

「这是我参与2022首次更文挑战的第23天,活动详情查看:2022首次更文挑战」。

题目

如果字符串中不含有任何 '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"
解释:这是该测试用例的唯一正确答案。

思路

比较经典的贪心的题目,类似的题目,只是换个描述的说法,还有很多。贪心可以理解为动态规划的一种特殊情况,局部最优解刚好也是全局最优解。
想要让字符串尽可能的长,其实就是尽量要把给出的字符用完,所以最自然而然的思路就是,每次都选择剩余数量最多的那个字符,除非触发限制条件出现了连续3次。
还可以这么考虑:给出abc字母的数量为ta、tb、tc,我们分别使用掉了ka、kb、kc个之后,分别剩余(ta-ka)、(tb-kb)、(tc-kc)个。我们当前的字符串长度为(ka+kb+kc),要使得总字符串最大,其实也就是要让剩余拼成的字符串最长,所以这里刚好局部最优跟全局最优是一致的。
有一个注意点,这里剩余数量最多是动态变化的,我们尽量使用当前剩余数量最大的字符,而不是一开始数量最多的字符。我们每用掉1个字符后,剩下的字符最多的可能会产生变化,所以每次用掉一个字符就要重新排序。

Java版本代码

public static String longestDiverseString(int a, int b, int c) {
    List<CharNum> list = new ArrayList();
    list.add(new CharNum('a', a));
    list.add(new CharNum('b', b));
    list.add(new CharNum('c', c));
    StringBuilder ans = new StringBuilder();
    int length = 0;
    while (true) {
        list = list.stream().sorted((o1, o2) -> o2.getNum() - o1.getNum()).collect(Collectors.toList());
        boolean breakFlag = false;
        for (CharNum item : list) {
            if (item.getNum() == 0) {
                breakFlag = true;
                break;
            }
            char ch = item.getC();
            if (length >= 2 && ch == ans.charAt(length-1) && ch == ans.charAt(length-2)) {
                continue;
            }
            ans.append(ch);
            length++;
            item.decrease();
            break;
        }
        if (breakFlag) {
            break;
        }
    }
    return ans.toString();
}

static class CharNum {
    private char c;
    private int num;

    public CharNum(char c, int num) {
        this.c = c;
        this.num = num;
    }

    public char getC() {
        return c;
    }

    public int getNum() {
        return num;
    }

    public void decrease() {
        num--;
    }
}