【Leetcode每日一题】621. 任务调度器（排序 可用堆/优先队列进行优化）

Leetcode 每日一题
题目链接： 621. 任务调度器
难度： 中等
解题思路： 对任务按照出现的次数进行排序，然后每次从出现次数最多的数开始，向后取n个，不够n的话补齐。 这个排序过程可以用堆或者是优先队列进行优化。本文未作优化。
题解：

class Solution:
    def leastInterval(self, tasks: List[str], n: int) -> int:
        # 对任务按照出现的次数进行排序
        ntasks = collections.Counter(tasks)
        ntasks = collections.OrderedDict(sorted(ntasks.items(), key = lambda x : x[1], reverse = True))

        # print(ntasks)

        # 出现次数最多的任务
        freq_task = list(ntasks.keys())[0]
        cnt = 0
        
        # 最后一个要减去的值
        tmp = 0

        while ntasks[freq_task] > 0:
            # print(ntasks)

            # 计数
            flag = 0

            for k, v in ntasks.items():
                if v == 0 or flag == n + 1:
                    break
                cnt += 1
                ntasks[k] -= 1
                flag += 1

            # 加上待命的值 最后一个不加, 在后面减去相应的值
            if flag <= n:
                tmp = n - flag + 1
                cnt += (n - flag + 1)

            # 这里可以用堆进行优化
            ntasks = collections.OrderedDict(sorted(ntasks.items(), key = lambda x : x[1], reverse = True))
            freq_task = list(ntasks.keys())[0]

        # 减去多加的
        cnt -= tmp
        
        # print(cnt)
        return cnt