621. 任务调度器

题目描述

给定一个用字符数组表示的 CPU 需要执行的任务列表。其中包含使用大写的 A - Z 字母表示的26 种不同种类的任务。任务可以以任意顺序执行,并且每个任务都可以在 1 个单位时间内执行完。CPU 在任何一个单位时间内都可以执行一个任务,或者在待命状态。

然而,两个相同种类的任务之间必须有长度为 n 的冷却时间,因此至少有连续 n 个单位时间内 CPU 在执行不同的任务,或者在待命状态。

你需要计算完成所有任务所需要的最短时间。

示例 :

输入:tasks = ["A","A","A","B","B","B"], n = 2
输出:8
解释:A -> B -> (待命) -> A -> B -> (待命) -> A -> B.
在本示例中,两个相同类型任务之间必须间隔长度为 n = 2 的冷却时间,而执行一个任务只需要一个单位时间,所以中间出现了(待命)状态。 

提示:

任务的总个数为 [1, 10000]。
n 的取值范围为 [0, 100]。

思路一:桶思想

思路参考:https://leetcode-cn.com/problems/task-scheduler/solution/tong-zi-by-popopop/

利用桶思想,先使用计数排序,对所有任务进行从大到小排序,

然后对计数数组进行排序

统计任务数最多的有几种任务

用公式计算:总排队时间 = (桶个数 - 1) * (n + 1) + 最后一桶的任务数(即最大桶的个数)

 1 class Solution {
 2     public int leastInterval(char[] tasks, int n) {
 3        // 利用桶思想
 4        // 先使用计数排序,对所有任务进行从大到小排序
 5        int len = tasks.length;
 6        int[] counts = new int[26];
 7        for(int i = 0; i < len; i++){
 8            counts[tasks[i] - 'A']++;
 9        }
10 
11        // 然后对计数数组进行排序
12        Arrays.sort(counts);
13 
14        // 统计任务数最多的有几种任务
15        int cnt = 1;
16        int N = counts[25];
17        for(int i = 24; i >= 0 && counts[i] == N; i--, cnt++);
18 
19        // 用公式计算
20        return Math.max((N - 1) * (n + 1) + cnt, len);
21     }
22 }

leetcode 运行时间为2ms, 空间为39.8MB

复杂度分析:

时间复杂度:计数排序的时间为O(n), 对计数数组排序的时间花费为O(26), 所以总的时间复杂度为O(n)

空间复杂度:O(1)