LeetCode 621. 任务调度器

题目描述

给定一个用字符数组表示的 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. 任务的总个数为 [1, 10000]。
2. n 的取值范围为 [0, 100]。

问题分析

创建一个大小为26的数组vec，用来统计给定数组tasks中各字符的个数。然后将vec按照由大到小排序，将vec[0]记为mx，这是出现次数最多的次数。然后从前往后遍历一下vec，统计出出现次数等于mx的字符有多少个，记为 i 。那么结果等于：模块数（mx-1）乘以每个模块的大小（n+1），再加上后缀 i 。最后还要把该结果与tasks数组的大小取较大的值来返回，这样做是用来处理一些特殊情况。

代码实现

class Solution {
public:
    int leastInterval(vector<char>& tasks, int n) {
        vector<int> vec(26, 0);
        for(char c : tasks)
            ++vec[c - 'A'];
        sort(vec.rbegin(), vec.rend());
        int mx = vec[0];
        int i = 0;
        while(vec[i] == mx)
            i++;
        int len = tasks.size();
        return max(len, (mx - 1) * (n + 1) + i);
    }
};