题目
给定一个用字符数组表示的 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); }