给定一个用字符数组表示的 CPU 需要执行的任务列表。其中包含使用大写的 A - Z 字母表示的26 种不同种类的任务。任务可以以任意顺序执行,并且每个任务都可以在 1 个单位时间内执行完。CPU 在任何一个单位时间内都可以执行一个任务,或者在待命状态。
然而,两个相同种类的任务之间必须有长度为 n 的冷却时间,因此至少有连续 n 个单位时间内 CPU 在执行不同的任务,或者在待命状态。
你需要计算完成所有任务所需要的最短时间。
示例 1:
输入: tasks = ["A","A","A","B","B","B"], n = 2 输出: 8 执行顺序: A -> B -> (待命) -> A -> B -> (待命) -> A -> B.
注:
任务的总个数为 [1, 10000]。
n 的取值范围为 [0, 100]。
C++
class Solution {
public:
int leastInterval(vector<char>& tasks, int n)
{
vector<int> tmp(26,0);
int m=tasks.size();
for(int i=0;i<m;i++)
{
tmp[tasks[i]-'A']++;
}
sort(tmp.begin(),tmp.end());
int val=tmp[25];
int count=0;
int i=25;
while(val==tmp[i] && i>=0)
{
count++;
i--;
}
int res=(n+1)*(val-1)+count;
return max(res,m);
}
};
python
class Solution:
def leastInterval(self, tasks, n):
"""
:type tasks: List[str]
:type n: int
:rtype: int
"""
m=len(tasks)
tmp=[0 for i in range(26)]
for i in range(m):
tmp[ord(tasks[i])-ord('A')]+=1
tmp.sort()
val=tmp[25]
count=0
i=25
while val==tmp[i] and i>=0:
count+=1
i-=1
res=(n+1)*(val-1)+count
return max(res,m)