题目
给定一个用字符数组表示的 CPU 需要执行的任务列表。其中包含使用大写的 A - Z 字母表示的26 种不同种类的任务。任务可以以任意顺序执行,并且每个任务都可以在 1 个单位时间内执行完。CPU 在任何一个单位时间内都可以执行一个任务,或者在待命状态。
然而,两个相同种类的任务之间必须有长度为 n 的冷却时间,因此至少有连续 n 个单位时间内 CPU 在执行不同的任务,或者在待命状态。
你需要计算完成所有任务所需要的最短时间。
示例 :
输入:tasks = ["A","A","A","B","B","B"], n = 2
输出:8
解释:A -> B -> (待命) -> A -> B -> (待命) -> A -> B.
在本示例中,两个相同类型任务之间必须间隔长度为 n = 2 的冷却时间,而执行一个任务只需要一个单位时间,所以中间出现了(待命)状态。
提示:
任务的总个数为 [1, 10000]。
n 的取值范围为 [0, 100]。
解题思路
因为每个任务之间的间隔必须得是n
,所以以n+1
个不同的任务为一轮,选出n + 1
个出现频率最高的任务,先安排这些任务,做完后把频率-1再放回到堆里。循环直到剩下的任务个数<= n + 1
为止。
当任务数不超过n + 1
时,此时完成任务的开销和出现频率最高的任务有关。设出现频率最高的任务有x
个,频率为xf
,则最终耗时应该是(n + 1) * (xf - 1) + x
看了外网大神的做法,最终的耗费时间,其实应该是任务数量task_num
+任务之间的空隙个数idles
,计算idles
的思路如下:
假设给出的任务是3A 2B 1C, n = 2
先放出现次数最多的A,得到 A # # A # # A
A分割出了count(A) - 1
个间隔,每个间隔中有n
个空隙,因此slot_num = n * (count(A) - 1)
把剩下的任务往这些空隙里填,计算剩下的任务个数,有available_task = task.length - count(A)
最终剩余的空隙个数为:idles = max(0, slot_num - available_task)
最后的返回值是task_num + idles
这是最理想的情况,有2种例外。
第1种例外:剩下的任务比空隙个数多,怎么填?
剩下的多余的任务,挤进空隙后面即可。比如3A 2B 1C, n = 1时,先用A分出来空隙:A # A # A,然后用B把两个#填上得到ABABA,轮到C时,没有空隙,直接挤进第一个AB后面,变成ABCABABA即可。
第2种例外,出现频率最高的任务有多个,比如3A 3B 1C, n = 2
同样用A先分割出来,得到A # # A # # A,然后用B贴着A填,得到A B # A B # A B,此时观察发现,把AB看作是X,其实就得到X # X # X,仍然符合理想情况的规律!不过此时空隙的个数会有一定的调整。
记最大频率的不同任务数为max_count
间隔个数没变,仍然是count(A) - 1
间隔中的空隙个数变了,变成:max(0, n - (max_count - 1))
可以插到空隙里的任务个数变了,变成:task.length - max_count * count(A)
所以如果有多个频率最大的不同任务,则会影响slot_num
,因此考虑了特殊情况后,有:
- 计算分割后的间隔个数:
sep_num = count(A) - 1
- 计算每个间隔中的空隙个数:
sep_slot_num = max(0, n - (max_count - 1))