任务调度器 中等🌟🌟🌟
课后作业
问题描述
原文链接：621. 任务调度器
给你一个用字符数组 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 <= 10的4次方
tasks[i] 是大写英文字母
n 的取值范围为 [0, 100]
代码实现
Java
class Solution {
// 出现次数最多的种类有多种，假设 k 种。
public int leastInterval(char[] tasks, int n) {
//先找出出现次数做多的字母
int[] arr = new int[26];
int k = 0;
for(int i = 0; i < tasks.length; i++){
arr[tasks[i] - 'A']++;
}
int max = 0;
for(int i = 0; i < 26; i++){
max = Math.max(max, arr[i]);
}
for(int i = 0; i < 26; i++){
if(arr[i] == max){
k++;
}
}
// 如果间隔够差
max = Math.max((n + 1) * (max - 1) + k, tasks.length);
// 如果间隔不够插
return max;
}
}
Python
class Solution(object):
def leastInterval(self, tasks, n):
"""
:type tasks: List[str]
:type n: int
:rtype: int
"""
# 找出出现次数最多的字母
arr = [0] * 26
k = 0
for task in tasks:
arr[ord(task) - ord('A')] += 1
max_count = 0
for count in arr:
max_count = max(max_count, count)
for count in arr:
if count == max_count:
k += 1
# 如果间隔够大
max_interval = max((n + 1) * (max_count - 1) + k, len(tasks))
return max_interval
C++
class Solution {
public:
int leastInterval(vector<char>& tasks, int n) {
// 找出出现次数最多的字母
vector<int> arr(26, 0);
int k = 0;
for (char task : tasks) {
arr[task - 'A']++;
}
int max_count = 0;
for (int count : arr) {
max_count = max(max_count, count);
}
for (int count : arr) {
if (count == max_count) {
k++;
}
}
// 如果间隔够大
int max_interval = max((n + 1) * (max_count - 1) + k, (int)tasks.size());
return max_interval;
}
};
Go
func leastInterval(tasks []byte, n int) int {
// 找出出现次数最多的字母
arr := make([]int, 26)
k := 0
for _, task := range tasks {
arr[task-'A']++
}
maxCount := 0
for _, count := range arr {
maxCount = max(maxCount, count)
}
for _, count := range arr {
if count == maxCount {
k++
}
}
// 如果间隔够大
maxInterval := max((n+1)*(maxCount-1)+k, len(tasks))
return maxInterval
}
func max(a, b int) int {
if a > b {
return a
}
return b
}