LeetCode 621 任务调度器

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

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

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

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

from typing import *
from collections import defaultdict, Counter
import heapq


class Solution:
    def leastInterval(self, tasks: List[str], n: int) -> int:
        last_idx = {}
        que = []
        cnts = Counter(tasks)
        for c in cnts:
            heapq.heappush(que, (-cnts[c], c))
        idx = 0
        while que or last_idx:
            last = idx - n - 1
            if last in last_idx:
                heapq.heappush(que, last_idx[last])
                del last_idx[last]
            if not que:
                idx += 1
            else:
                t = heapq.heappop(que)
                if t[0] + 1 < 0:
                    last_idx[idx] = (t[0] + 1, t[1])
                idx += 1
        return idx