算法学习笔记 2024/04/17

今天继续学习栈和队列

1. 滑动窗口的最大值

这道题用O(n)的复杂度，应该使用队列。普通的队列可以实现区间的移动，但是要实现最大值的查找，还需要对队列元素的进入顺序进行调控。

使用deque实现一个单调队列，使得进入的最大元素始终在队列顶，注意deque的语法，deque是一个双开口的队列，定义时要注意弹出的端口。

class Solution {
private:
    class Deque{
    public:
        deque<int> deque;
        void pop(int value){
            if( !deque.empty() && value == deque.front()){
                deque.pop_front();
            }
        }
        void push(int value){
            while( !deque.empty() && value > deque.back()){
                deque.pop_back();
            }
            deque.push_back(value);
        }
        int front(){
            return deque.front();
        }
    };
public:
    vector<int> maxSlidingWindow(vector<int>& nums, int k) {
        vector<int> result;
        Deque que;
        for( int i = 0; i < k; i++){
            que.push(nums[i]);
        }
        result.push_back(que.front());
        int j = k;
        while(j < nums.size()){
            que.pop(nums[j-k]);
            que.push(nums[j]);
            result.push_back(que.front());
            j++;
        }
        return result;
    }
};

题例： 239. 滑动窗口最大值

2. 前K个高频元素

这道题目需要使用小顶堆的结构，去存储前k个最大值，最大堆在更新时需要弹出最大值，反而无法存储前k大的值，而最小堆能弹出最小值来维护前k大的元素。首先需要一个哈希表来记录数组元素出现的频率。

STL库函数定义优先队列的函数：

对常规的int数据类型

最大优先队列：priority_queue<int，vector<int>, less<int> > MaxHeap

最小优先队列：priority_queue<int，vector<int>，greater<int> > MinHeap

对于结构体等复杂数据结构，需要定义一个比较大小的双目运算符。注意双目运算符的重载应该在class外。

对于索引的数据类型，本题中比较的是哈希map中nums[i]的频数，需要自定义比较类型。决定优先级的是pair<iint,int>中的second。

class mycomparison {
    public:
        bool operator()(const pair<int, int>& lhs, const pair<int, int>& rhs) {
            return lhs.second > rhs.second;
        }
    };

创建priority_queue时，构造函数中的比较类使用自己定义的类

class Solution {
public:
    class PriorityQue{
        public:
        bool operator()(const pair<int,int>& lhs,const pair<int,int>& rhs){
            return lhs.second > rhs.second;
        }
    };
    vector<int> topKFrequent(vector<int>& nums, int k) {
        unordered_map<int,int> map;
        for(int i = 0; i < nums.size(); i++){
            map[nums[i]]++;
        }
        priority_queue<pair<int,int>, vector <pair<int,int>> ,PriorityQue> minheap;
        for(unordered_map<int,int>::iterator it = map.begin(); it!=map.end(); it++){  //迭代器it遍历哈希map
            minheap.push(*it);
            if(minheap.size()>k){
                minheap.pop();
            }
        }
        vector <int> result(k);
        for(int i = k-1; i >=0 ; i--){
            result[i] = minheap.top().first;
            minheap.pop();
        }
        return result;
    }
};

题例： 347. 前 K 个高频元素