编程训练第一百二十二期——任务调度器

编程问题：

给你一个用字符数组 tasks 表示的 CPU 需要执行的任务列表。其中每个字母表示一种不同种类的任务。任务可以以任意顺序执行，并且每个任务都可以在 1 个单位时间内执行完。在任何一个单位时间，CPU 可以完成一个任务，或者处于待命状态。

然而，两个 相同种类 的任务之间必须有长度为整数 n 的冷却时间，因此至少有连续 n 个单位时间内 CPU 在执行不同的任务，或者在待命状态。你需要计算完成所有任务所需要的 最短时间 。

---

示例：

• 输入：tasks = [“A”,“A”,“A”,“B”,“B”,“B”], n = 2
  输出：8
  解释：A -> B -> (待命) -> A -> B -> (待命) -> A -> B
  在本示例中，两个相同类型任务之间必须间隔长度为 n = 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”]
  …
  诸如此类
• 输入：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.桶思想
记录最大任务数量 N，看一下任务数量并列最多的任务有多少个，即最后一个桶子的任务数 X，计算 NUM1=(N-1)*(n+1)+X,计算无空闲的时执行的任务数NUM2=tasks.size()，两者输出其中较大值即可，因为存在空闲时间时肯定是 NUM1 大，不存在空闲时间时肯定是 NUM2>=NUM1。

时间复杂度 O(MlogM)，其中M是任务的种类数目
空间复杂度 O(1)

class Solution {
private:
    static bool compare(const int& a, const int& b)
    {
        return a > b;
    }

public:
    int leastInterval(vector<char>& tasks, int n) {
        int len = tasks.size();
        vector<int> tmp(26);
        for (auto x : tasks)
            tmp[x - 'A']++;
        sort(tmp.begin(), tmp.end(), compare);
        int cnt = 1;
        while(cnt < tmp.size() && tmp[cnt] == tmp[0])
            cnt++;
        return max(len, cnt+(n+1)*(tmp[0]-1));
    }
};