[LeetCode]621. 任务调度器(贪心)

Posted 2021-03-25 今天GaGa打代码了吗？

题目

给定一个用字符数组表示的 CPU 需要执行的任务列表。其中包含使用大写的 A - Z 字母表示的26 种不同种类的任务。任务可以以任意顺序执行，并且每个任务都可以在 1 个单位时间内执行完。CPU 在任何一个单位时间内都可以执行一个任务，或者在待命状态。

然而，两个相同种类的任务之间必须有长度为?n 的冷却时间，因此至少有连续 n 个单位时间内 CPU 在执行不同的任务，或者在待命状态。

你需要计算完成所有任务所需要的最短时间。

示例 1：

输入: tasks = ["A","A","A","B","B","B"], n = 2
输出: 8
执行顺序: A -> B -> (待命) -> A -> B -> (待命) -> A -> B.
注：

任务的总个数为?[1, 10000]。
n 的取值范围为 [0, 100]。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/task-scheduler
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题解

1 只需找出出现次数最多的字符，假设其出现次数为maxOccurTimes，把它们间隔为n的放置，其余task（或空）补足间隔即可；易得字符数组长度为(maxOccurTimes - 1) * (n + 1) + 1； （贪心的思想）
2 但考虑出现次数等于最大次数的字符有多个为maxOccurKind，那么这几个字符排在字符数组末尾，字符数组长度应改为（maxOccurTimes - 1) * (n - 1) + maxOccurKind;
3 若上式结果<原字符数组长度，说明按上述方法间隔里排不下，那么应取原字符数组长度。
4 若最大次数的字符种数maxOccurKind>间隔数，那么在原字符数组长度一定可以排下，也应取原字符数组长度。

代码

public static int leastInterval(char[] tasks, int n) {
        if (tasks == null) {
            return 0;
        }

        HashMap<Character, Integer> occurTimesMap = new HashMap<>();
        for (char task : tasks) {
            occurTimesMap.merge(task, 1, Integer::sum);
        }

        int maxOccurTimes = 0;// 任务的最大次数
        int maxOccurKind = 0;// 次数等于最大次数的任务数
        for (int occurTimes : occurTimesMap.values()) {
            if (occurTimes > maxOccurTimes) {
                maxOccurTimes = occurTimes;
                maxOccurKind = 1;
            } else if (occurTimes == maxOccurTimes) {
                maxOccurKind++;
            }
        }

        if (maxOccurKind > maxOccurTimes) {
            return tasks.length;
        }

        return Math.max((maxOccurTimes - 1) * (n + 1) + maxOccurKind, tasks.length);
    }