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. 任务的总个数为 [1, 10000]。
2. n 的取值范围为 [0, 100]。

思路

• 先安排数量多的任务
• 比如，n取2
  
  
• 梳理一下思路，先排最多的任务A，假设A任务有k个，冷却时间为n，那么排完前k-1个A任务必定会占用(k-1)*(n+1)的时间。然后最后一个A任务必定会占用一个时间。
• 因为A是最多的，其他任务都小于等于A，最后一个A后面，接着放和A数量一样多的任务。如果最终计算出来的时间小于总的任务数，那说明任务可以排满，返回任务数即可。

代码

class Solution {
public:
    int leastInterval(vector<char>& tasks, int n) {
        if ( n == 0 ) return tasks.size();

        int res = 0;

        int memo[26] = { 0 };

        for ( auto& c : tasks )
            ++memo[c-'A'];
        
        sort( memo, memo+26);

        res += ( memo[25] - 1 ) * ( n + 1 ) + 1;
        for ( int i = 24; i >= 0 && memo[i] == memo[25]; --i )
            ++res;
        
        return res > tasks.size() ? res : tasks.size();
    }
};