LeetCode-621. 任务调度器

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. 任务的总个数为 [1, 10000]。
2. n 的取值范围为 [0, 100]。

解题思路：统计每种任务的数量，计max_count数量最大值，计num_max_count为任务数量等于max_count的任务种类数，那么最短时间=max((max_count-1)*(n+1)+num_max_count,len(tasks))。

原理解释如下：

假设A为tasks中数量最大的任务种类，n为2，那么执行顺序为AXXAXXAXX...A，其中X代表其他任务种类或者待命状态间隔插空，即可在最短时间内完成所有任务的调度。

假设A，B同为tasks中数量最大的任务种类，n为2，那么执行顺序为ABXXABXXABXX...AB，其中X代表其他任务种类或者待命状态间隔插空，即可在最短时间内完成所有任务的调度。

为什么要对len(tasks)取较大值呢？因为当n较小时，比如0，num_max_count较大时，比如2，n+1可能就不能表示每段任务序列的长度。例子：[A,AA,B,B,B]，n = 0，调度顺序为ABABAB，按照公式计算(3-1)*(0+1)+2 = 4，其中(0+1)不能表示任务子序列的长度。

Python3代码如下：

class Solution(object):
    def leastInterval(self, tasks, n):
        """
        :type tasks: List[str]
        :type n: int
        :rtype: int
        """
        d = collections.Counter(tasks)
        max_count = 0
        for key in d:
            max_count = max(max_count,d[key])
        num_max_count = 0
        for key in d:
            if d[key] == max_count:
                num_max_count += 1
        return max((max_count-1)*(n+1)+num_max_count,len(tasks))