力扣 621. 任务调度器 贪心/思维

https://leetcode-cn.com/problems/task-scheduler/

思路一：贪心。考虑到$n$的冷却时间，我们自然想到优先执行任务数较多的任务，所以需要把不同种类的任务按照任务数量从大到小排序，每一轮可以选取前$n+1$个任务执行，然后统计时间即可。

class Solution {
public:
    int leastInterval(vector<char>& tasks, int n) {
        int cnt[26]={0};
        int siz=tasks.size();
        for(int i=0;i<siz;i++)
            ++cnt[tasks[i]-'A'];
        auto func=greater<int>();
        sort(cnt,cnt+26,func);
        int time=0;
        while(cnt[0]){
            int num=min(26,n+1),i;
            for(i=0;i<num;i++){
                if(!cnt[i])
                    break;
                --cnt[i];
                ++time;
            }
            //还有任务没有完成
            //但是这一轮的工作不足n+1个 需要加上额外的时间
            if(cnt[0])
                time+=n-i+1;
            sort(cnt,cnt+26,func);
        }
        return time;
    }
};

思路二：看官方题解学到的，非常巧妙，看了几遍才理解…估计下次碰到还是想不起来orz。


其实他最后说的有点抽象，看图自己理解会好一点。如果剩余的时间没有用完，那么这些时间会浪费到待命上，所以总时间等于待命时间+任务数量，如果用完的话，说明没有待命时间，那么总时间就等于任务数量。

class Solution {
public:
    int leastInterval(vector<char>& tasks, int n) {
        int cnt[26]={0};
        int siz=tasks.size();
        for(int i=0;i<siz;i++)
            ++cnt[tasks[i]-'A'];
        auto func=greater<int>();
        sort(cnt,cnt+26,func);
        int MAX=cnt[0]-1;
        int time=MAX*n;
        for(int i=1;i<26&&cnt[i];i++)
            time-=min(MAX,cnt[i]);
        time=time>0?time+siz:siz;
        return time;
    }
};