leetcode 621. Task Scheduler

最新推荐文章于 2024-01-17 22:26:13 发布

原创最新推荐文章于 2024-01-17 22:26:13 发布 · 386 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 专栏收录该内容

280 篇文章

订阅专栏

本文介绍了一种基于优先队列的任务调度算法，该算法用于解决CPU在处理任务时考虑到冷却间隔n的任务分配问题。通过使用优先队列按任务出现频率排序，算法能够在满足冷却间隔限制的情况下，尽可能减少完成所有任务所需的总时间。

621. Task Scheduler

Given a char array representing tasks CPU need to do. It contains capital letters A to Z where different letters represent different tasks.Tasks could be done without original order. Each task could be done in one interval. For each interval, CPU could finish one task or just be idle.

However, there is a non-negative cooling interval n that means between two same tasks, there must be at least n intervals that CPU are doing different tasks or just be idle.

You need to return the least number of intervals the CPU will take to finish all the given tasks.

Example 1:

Input: tasks = ["A","A","A","B","B","B"], n = 2
Output: 8
Explanation: A -> B -> idle -> A -> B -> idle -> A -> B.

Note:

The number of tasks is in the range [1, 10000].
The integer n is in the range [0, 100].

1、贪心，利用优先队列排序：队列中保存 <个数，类型> 的map，并且按照个数由大到小排序。按照词频由大到小取出n+1个或者队列中全部(若没有取出全部，则总长度要加上空闲个数)，再把词频-1之后不为0的放回队列中。直到队列空了为止。——其实也和前面相同，总是选择词频最大的填入每一块。

2、priority_queue 取出第一个是 .top(); 而不是 front();

3、priority_queue 自定义类型需要重载操作符。和set、map的重载不太一样。

bool operator<(pair<int, char> a, pair<int, char> b)
{
    return a.first > b.first;
}

class Solution {
public:
    int leastInterval(vector<char>& tasks, int n) 
    {
        if (n == 0)  return tasks.size();
        map<char, int> mp;
        for (auto it : tasks)
            mp[it]++;

        priority_queue<pair<int, char>> pq;
        for (auto it : mp)
            pq.push(make_pair(it.second, it.first));
        
        int ret = 0;
        queue<pair<int, char>> tmp;
        while (!pq.empty())
        {
            int k = n + 1;
            while (k--)
            {
                if (!pq.empty())
                {
                    auto it = pq.top();
                    pq.pop();  
                    it.first--;
                    if (it.first > 0)
                        tmp.push(it);
                }
                else if (tmp.empty())   //如果pq空了 tmp也空了 说明任务都完毕了
                    return ret;
                ret++;
            }
            
            //把tmp装回去
            while (!tmp.empty())
            {
                pq.push(tmp.front());
                tmp.pop();
            }
        }
        return ret;
    }
};