LeetCode 中级 - Task Scheduler

Task Scheduler

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

分析

笔者在这里采取了一个比较笨重的方法，将任务按照频次排序，以CD长度为一个循环，每次按照频次降序插入一个任务，循环往复，直到所有任务都执行完成。

网上还给出了一种比较偏向数学理解的方法。

取最大的频次假如是4次A，则将全部任务分成4块，每一块长度大于等于cd，加入cd为3，大概如下

A—A—A—A—，向其中插入元素，可满足每一个任务都插入其中。

具体可自行实践。

代码

    class Solution {
        public int leastInterval(char[] tasks, int n) {
            int cd=n+1;
            int res=0;
            PriorityQueue<Integer> queue = new PriorityQueue<>((a,b)->b-a);

            HashMap<Character,Integer> map = new HashMap<>();
            for(int i=0;i<tasks.length;i++){
                map.put(tasks[i], map.getOrDefault(tasks[i], 0) + 1); 

            }

            queue.addAll(map.values());

            while(!queue.isEmpty()){
                List<Integer> tmp = new ArrayList<>();
                int time = 0;
                for(int i=0;i<cd;i++){
                    if(!queue.isEmpty()){
                        tmp.add(queue.peek());
                        queue.poll();
                        time++;
                    }
                }

                for(int i:tmp){
                    if(i>1){
                        queue.add(i-1);
                    }
                }
                //System.out.println(""+queue.toString());
                if(queue.isEmpty()){
                    res+=time;
                }else{
                    res+=cd;
                }
            }
            return res;
        }
    }