目录
一、题目
给你一个用字符数组 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：
扫描二维码关注公众号，回复： 12472068 查看本文章
输入：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]
三、思路
- 1、将任务进行归类，并记录任务的个数；
- 2、按照任务的个数将任务进行排列；
- 3、如果空位都插满之后还有任务没有排布，则随便插入即可。
四、代码
class Solution:
def leastInterval(self, tasks, n: int) -> int:
"""
:param tasks: List[str]
:param n: int
:return: int
"""
"""
解题思路：
1、将任务进行归类，并记录任务的个数
2、按照任务的个数将任务进行排列
3、如果空位都插满之后还有任务没有排布，则随便插入即可。
"""
if len(tasks) <= 1 or n < 1:
return len(tasks)
# 将任务进行归类，并记录任务的个数
counts = [0] * 26
for i in tasks:
counts[ord(i) - 65] += 1
print(counts)
# 按照任务的个数将任务进行排列
"""
maxCount: 任务的最大个数
retCount: 第一个任务的排布
"""
counts.sort()
print(counts)
maxCount = counts[25]
# A - () - () - A - () - () - A
# retCount = 2 * [A - () - ()] + A
retCount = (maxCount - 1) * (n + 1) + 1
# 排布其他任务
i = 24
while i >= 0 and counts[i] == maxCount:
# A - B - () - A - B - () - A - B
retCount += 1
i -= 1
return max(retCount, len(tasks))
if __name__ == '__main__':
tasks = ["A", "A", "A", "B", "B", "B"]
n = 2
s = Solution()
ans = s.leastInterval(tasks, n)
print(ans)