基础知识
题目:
给你一个用字符数组 tasks 表示的 CPU 需要执行的任务列表。其中每个字母表示一种不同种类的任务。任务可以以任意顺序执行,并且每个任务都可以在 1 个单位时间内执行完。在任何一个单位时间,CPU 可以完成一个任务,或者处于待命状态。然而,两个 相同种类 的任务之间必须有长度为整数 n 的冷却时间,因此至少有连续 n 个单位时间内 CPU 在执行不同的任务,或者在待命状态。你需要计算完成所有任务所需要的 最短时间 。
思路:
桶思想:https://leetcode-cn.com/problems/task-scheduler/solution/tong-si-xiang-jian-ji-gao-xiao-by-hzhu212/
补充说明:
初步计算总排队时间,并将其计为res:
桶的个数 是重复次数最多的任务的 个数,计为m,桶的大小为n+1。
但所有这些桶 不一定能将所有任务装下。若任务个数>m*(n+1),则该桶不能将所有元素装下。但可以通过扩充桶大小将所有这些任务装进去,且一定能够满足 所给的冷却时间 要求,此时所需时间为task总数。若任务个数<=m*(n+1),则所给的桶 即可将所有元素装下,此时所需时间为(m-1)*(n+1)+last_bucket_size.
解答:
class Solution:
def leastInterval(self, tasks: List[str], n: int) -> int:
ct = Counter(tasks)
# m为出现次数最多的任务 的个数,即所需的桶 的个数
m = ct.most_common(1)[0][1]
last_bucket_size = list(ct.values()).count(m)
res = (m- 1) * (n + 1) + last_bucket_size
return max(res, len(tasks))