621-task-scheduler.md

621. 任务调度器 - medium

给你一个用字符数组 tasks 表示的 CPU 需要执行的任务列表。其中每个字母表示一种不同种类的任务。任务可以以任意顺序执行，并且每个任务都可以在 1 个单位时间内执行完。在任何一个单位时间，CPU 可以完成一个任务，或者处于待命状态。

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

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

 

示例 1：

输入：tasks = ["A","A","A","B","B","B"], n = 2
输出：8
解释：A -> B -> (待命) -> A -> B -> (待命) -> A -> B
     在本示例中，两个相同类型任务之间必须间隔长度为 n = 2 的冷却时间，而执行一个任务只需要一个单位时间，所以中间出现了（待命）状态。 

示例 2：

输入：tasks = ["A","A","A","B","B","B"], n = 0
输出：6
解释：在这种情况下，任何大小为 6 的排列都可以满足要求，因为 n = 0
["A","A","A","B","B","B"]
["A","B","A","B","A","B"]
["B","B","B","A","A","A"]
...
诸如此类

示例 3：

输入：tasks = ["A","A","A","A","A","A","B","C","D","E","F","G"], n = 2
输出：16
解释：一种可能的解决方案是：
     A -> B -> C -> A -> D -> E -> A -> F -> G -> A -> (待命) -> (待命) -> A -> (待命) -> (待命) -> A

 

提示：

• 1 <= task.length <= 104
• tasks[i] 是大写英文字母
• n 的取值范围为 [0, 100]

Solutions

构造一个调用图。

找到调用次数最多的任务，次数记为 maxv，如果只有这个任务，则需要 (maxv - 1) * (n + 1) + 1 单位的时间。如果存在 maxc 个 maxv 次调用的任务。则需要 (maxv - 1) * (n + 1) + maxc 单位的时间。

假设剩下的任务数小于右侧的空闲时间。则这就是最短时间。如果剩下任务数大于右侧空闲时间数。那就可以不需要空闲时间。最短时间等于总的任务个数。

class Solution:
    def leastInterval(self, tasks: List[str], n: int) -> int:
        maxv, maxc = 0, 0
        m = {}
        for t in tasks:
            tv = m.get(t, 0)
            tv += 1
            if tv == maxv:
                maxc += 1
            elif tv > maxv:
                maxc = 1
                maxv = tv
            m[t] = tv
        
        n = (maxv - 1) * (n + 1) + maxc
        return max(n, len(tasks))