LeetCode 621. 任务调度器（C++、python）

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

C++

class Solution {
public:
    int leastInterval(vector<char>& tasks, int n) 
    {
        vector<int> tmp(26,0);
        int m=tasks.size();
        for(int i=0;i<m;i++)
        {
            tmp[tasks[i]-'A']++;
        }
        sort(tmp.begin(),tmp.end());
        int val=tmp[25];
        int count=0;
        int i=25;
        while(val==tmp[i] && i>=0)
        {
            count++;
            i--;
        }
        int res=(n+1)*(val-1)+count;
        return max(res,m);
    }
};

python

class Solution:
    def leastInterval(self, tasks, n):
        """
        :type tasks: List[str]
        :type n: int
        :rtype: int
        """
        m=len(tasks)
        tmp=[0 for i in range(26)]
        for i in range(m):
            tmp[ord(tasks[i])-ord('A')]+=1
        tmp.sort()
        val=tmp[25]
        count=0
        i=25
        while val==tmp[i] and i>=0:
            count+=1
            i-=1
        res=(n+1)*(val-1)+count
        return max(res,m)