目录
一、题目描述
给你一个用字符数组 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:
输入: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]
二、解题思路
这题妙就妙在思维,不要被题目的描述限制住。
只有两种情况,一种是任务种类少于等于n+1时,此时可以想象成如下图中的情况(图来自力扣题解区,侵删),如5个A,5个B,2个C, n为2:
把(n+1)*(max - 1)+ maxCount想象成一个任务的容器,在这种情况下可能存在冷却时间,如上图(任务数少于容器大小时);也可能不存在冷却时间(任务数等于容器大小时),比如当C的个数为5时。但执行的时间始终等于容器的大小。
另一种情况为任务种类数大于n+1时,如5个A,5个B,3个C,2个D,2个E,2个F, 1个G, n为2:
这种情况下因为任务种类数多于n+1,所以只要把不同的任务挨着执行,就很有可能不存在冷却时间。同样有两种情况:任务数少于容器大小时,因为任务数太少,容器填不满,还是需要冷却时间,执行时间为容器的大小;任务书大于容器大小时,此时容器装不下,那么把不同任务挨着执行,就不会有冷却时间,此时的执行时间就是任务的数量。
综合一下:执行时间=容器的大小与任务数两者的最大值。
三、代码实现
#include <bits/stdc++.h>
using namespace std;
int leastInterval(vector<char>& tasks, int n) {
vector<int> taskTimes(26);
for (auto c : tasks) {
taskTimes[c - 'A']++;
}
int maxTimes = 0;
for (auto x : taskTimes) {
maxTimes = max(x, maxTimes);
}
int maxTaskCount = 0;
for (auto x : taskTimes) {
if (x == maxTimes)
maxTaskCount++;
}
int size = tasks.size();
return max((n + 1)* (maxTimes - 1) + maxTaskCount, size);
}
int main() {
vector<char> tasks = { 'A','A','A','A','A','B','B','B','B','B','C','C','C' };
cout << leastInterval(tasks, 2) << endl;
return 0;
}