给定一个用字符数组表示的 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]。
思路一:
首先考虑频率最高的字符,我们可以先确定它们的相对位置,然后将它们用作边界墙来插入剩余的频率低的字符。以下是证明:
设F是频率为k的最频繁的字符集。
我们可以创建k个块,每个块是相同的,并且是由特定固定顺序的F中的字符组成的字符串。
让这些块的头部为H_i; 那么H_2的频率肯定小于等于H_1; 然后我们将频率较低的字符插入到这些块之间的间隙中,逐个按递减顺序逐个排列,并尝试在每次插入字符时尽可能完整或均匀地填充k-1间隙。总之,将较不频繁的字符顺序地附加到第一个k-1块的每个块的末尾,然后加入块并保持它们的头部彼此之间的相对距离至少为n。
例子:
AAAABBBEEFFGG 3
这里X代表一个空隙:
Frame: "AXXXAXXXAXXXA"
insert 'B': "ABXXABXXABXXA" <--- 'B' 比其他字符有更高的频率,先插入它.
insert 'E': "ABEXABEXABXXA"
insert 'F': "ABEFABEXABFXA" <--- 每次都尽量填充剩余k-1个空隙.
insert 'G': "ABEFABEGABFGA"
AACCCBEEE 2
3 个独立的块 "CE", "CE CE CE" <-- 把 "CE" 当做一个块
插入'A' 在空隙中,因为他比 'B' 有更高的频率---> "CEACEACE"
插入'B' ---> "CEABCEACE" <----- 结果是 tasks.length;
AACCCDDEEE 3
3 个独立块 "CE", "CE CE CE" <--- 同理,"CE" 当成一个块.
首先插入'A'->"CEA CEA CE"
然后插入 'B'->"CEABCEABCE" <---- 结果是 tasks.length;
ACCCEEE 2
3 个独立块 "CE", "CE CE CE" <-- 同理,"CE" 当成一个块.
首先插入 'A' --> "CEACE CE" <-- 结果是 (c[25] - 1) * (n + 1) + 25 -i = 2 * 3 + 2 = 8
最后公式(c[25] - 1) * (n + 1) + 25 -i
的意思是:每个块的长度为(n + 1)
(不包含最后一个块),一共有(c[25] - 1)个长度为(n + 1)的块,再加上最后一个块的长度为25-i。
参考代码:
class Solution {
public:
int leastInterval(vector<char>& tasks, int n) {
map<char, int> m;
for (auto task : tasks) m[task]++;
vector<int> count(26, 0);
for (auto it = m.begin(); it != m.end(); it++) count[it->first - 'A'] = it->second;
sort(count.begin(), count.end());
int i = 25;
while (i >= 0 && count[i] == count[25]) i--;
return max((int)tasks.size(),(count[25] - 1)*(n + 1) + (25 - i));
}
};
思路二:这道题首先统计字符的次数,然后和上面的解法一样,对于每个快,应该有n+1的长度,设置cycle=n+1,那么每次就来填充cycle个长度的元素。那么具体怎么做呢?把统计的字符的个数加到最大堆中,每次取堆的top cycle个元素,取出的元素对应的次数减一,如果不为0就继续加到堆中。按照道理说我们每次都对结果加cycle次,但是有特例,就像上面的方法倒数第二个例子,会单出最后的一个CE块,不满足CE的长度为3,所以在取top cycle个元素的时候需要有个计数器,最后判断此时堆是否为空,若为空表示到了最后一个块,结果加计数器的值,否则加cycle。
参考代码:
class Solution {
public:
int leastInterval(vector<char>& tasks, int n) {
map<char, int> m;
int res = 0,cycle=n+1;
for (auto task : tasks) m[task]++;
priority_queue<int> q;
for (auto it = m.begin(); it != m.end(); it++) q.push(it->second);
while (!q.empty()) {
vector<int> tmp;
int cnt = 0;
for (int i = 0; i < cycle; i++) {
if (!q.empty()) {
tmp.push_back(q.top()); q.pop();
cnt++;
}
}
for (auto t : tmp) {
if (--t) q.push(t);
}
res += q.empty() ? cnt : cycle;
}
return res;
}
};