Leetcode C++《热题 Hot 100-62》621. 任务调度器

Leetcode C++《热题 Hot 100-62》621. 任务调度器

1. 题目

给定一个用字符数组表示的 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]。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/task-scheduler
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 思路

• 贪心思路，n个单位时间中每次选择最多次的前n个任务去执行，如果不够n则空过。最后当所有任务执行完之后，结束时间即为所需时间

3. 代码

class Solution {
public:
    int leastInterval(vector<char>& tasks, int n) {
        //只有26种任务到是
        vector<int> taskTimes;
        for (int i = 0; i < 26; i++)
            taskTimes.push_back(0);
        for (int i = 0; i < tasks.size(); i++) {
            taskTimes[int(tasks[i]-'A')]++;
        }
        sort(taskTimes.begin(), taskTimes.end(), greater<int>());
        int allTime = 0;
        while(true) {
            int k = 0;
            for (int i = 0; i < n+1; i++) {
                if (taskTimes[i] > 0) {
                    taskTimes[i]--;
                    k = i;
                } else {
                    break;
                }        
            }
            sort(taskTimes.begin(), taskTimes.end(),greater<int>());
            if (taskTimes[0] == 0) {
                return allTime+k+1;
            } else {
                allTime += (n+1);
            }
        }
    }
};