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

题目

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