给定一个用字符数组表示的 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]。
代码
class Solution {
public:
int leastInterval(vector<char>& tasks, int n) {
vector<int> cnt(26, 0);
for (char task : tasks) {
++cnt[task - 'A'];
}
sort(cnt.begin(), cnt.end());
int i = 25, mx = cnt[25], len = tasks.size();
while (i >= 0 && cnt[i] == mx) --i;
return max(len, (mx - 1) * (n + 1) + 25 - i);
}
};
算法
记任务总个数是len
个,单类任务最多k
个,k
个任务有m
种,间隔是n
。给出答案是max(len, (m-1)*(n+1)+k)
首先是排序
k
类的任务,这些任务单个具有个数是m
n+1<=k
,则剩下种类的任务在m-1
个块间隔中可以插入。因为它们最多单个任务由m-1
次。这样时间是t=len>=(m-1)*(n+1)+k
n+1>k
,每个块剩下n+1-m
个空白位置可以插入。len<=(m-1)*(n+1)+k
,空白位置个数是不少于剩下的任务个数的,那么任务总是可以利用这些空白位置插入的。比如按照类别的任务个数从多到少进行,插入从空白位置多的间隔到空白位置少的间隔。这样,各个间隔的空间余量最多相差1。len>(m-1)*(n+1)+k
,空白位置个数是<
剩下的任务个数的。可以看作有m
个间隔,最后一个间隔的空间余量是len-(m-1)*(n+1)-k
。按照上面的方法进行排序即可