一.题目:
CPU任务调度问题,同样的两个task之间的时间间距不能少于n,否则会用idle填充。问这批task需要多少时间片才能完成。
Example:
Input: tasks = ["A","A","A","B","B","B"], n = 2
Output: 8
Explanation: A -> B -> idle -> A -> B -> idle -> A -> B.
二.解题思路:
首先我们统计每种任务出现的次数.我们需要最少的任务时间:(最多任务数-1)*(n + 1) + (相同最多任务的任务个数)。如果n比较大,我们用上述公式就能得到最少的任务时间,因为剩余的任务在不违背要求间隔的情况下穿插进间隔位置即可,空缺位置补idle。但是当n很小的时候可能会导致上述公式计算的任务时间甚至比总任务长度还短,所以需要取两者的较大值。
代码如下:
class Solution(object):
def leastInterval(self, tasks, n):
"""
:type tasks: List[str]
:type n: int
:rtype: int
"""
count = collections.Counter(tasks).most_common()
max_cnt = count[0][1]
max_same_cnt = 0
for i in count:
if i[1] == max_cnt:
max_same_cnt +=1
return max(len(tasks), (max_cnt-1) * (n+1) + max_same_cnt)