代码随想录第13天|栈与队列

239. 滑动窗口最大值

补充: deque的用法

#include <iostream>
#include <deque>

int main() {
    std::deque<int> myDeque;

    // 在后端插入元素
    myDeque.push_back(5);
    myDeque.push_back(2);
    // 在前端插入元素
    myDeque.push_front(8);
    
    // 删除末尾元素
    myDeque.pop_back();
    // 删除首部元素
    myDeque.pop_front();
    
    // 访问首尾元素
    std::cout << "Front element: " << myDeque.front() << std::endl;
    std::cout << "Back element: " << myDeque.back() << std::endl;

	// 遍历 deque
    for (auto it = myDeque.begin(); it != myDeque.end(); ++it) {
        std::cout << *it << " ";
    }
    std::cout << std::endl;
    
    // 在指定位置插入元素
    std::deque<int>::iterator it = myDeque.begin();
    it++;
    myDeque.insert(it, 3);

    // 检查 deque 是否为空
    if (myDeque.empty()) {
        std::cout << "Deque is empty" << std::endl;
    } else {
        std::cout << "Deque is not empty" << std::endl;
    }

    return 0;
}

优先队列:

• 比如 1 3 -1时, 大顶堆的值为 3 1 -1
• 将 -3 移至里面, 1 移除, 无法pop元素1
• 通过特殊方法, 用pair将下标加入
  参考

class Solution {
public:
    vector<int> maxSlidingWindow(vector<int>& nums, int k) {
        int n = nums.size();
        priority_queue<pair<int, int>> q;
        for (int i = 0; i < k; ++i) {
            q.emplace(nums[i], i);
        }
        vector<int> ans = {q.top().first};
        for (int i = k; i < n; ++i) {
            q.emplace(nums[i], i);
            while (q.top().second <= i - k) {
                q.pop();
            }
            ans.push_back(q.top().first);
        }
        return ans;
    }
};

单调队列

• 单调递增或单调递减
• 队列没有必要维护窗口里的所有元素, 只需要维护有可能成为窗口里最大值的元素, 保证元素从大到小的排列
• 原则
  • push新元素, 若比队列前面的元素都大, 则pop, 直至最前面的元素大于新进的元素为止

参考

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

---

347.前 K 个高频元素
给你一个整数数组 nums 和一个整数 k ，请你返回其中出现频率前 k 高的元素。你可以按 任意顺序 返回答案

补充: 大根堆的用法

思路: 最小堆

class Solution {
public:
    struct cmp {
        bool operator()(const pair<int, int>& a, const pair<int, int>& b) {
            return a.second > b.second;
        }
    };
    vector<int> topKFrequent(vector<int>& nums, int k) {
        map<int, int> mymap;
        for (int i = 0; i < nums.size(); i++) {
            mymap[nums[i]]++;
        }
        priority_queue<pair<int, int>, vector<pair<int, int>>, cmp> myqueue;
        for (auto it = mymap.begin(); it != mymap.end(); it++) {
            if (myqueue.size() < k) {
                myqueue.push(*it);
            } else {
                myqueue.push(*it);
                myqueue.pop();
            }
        }
        vector<int> result;
        while (!myqueue.empty()) {
            result.push_back(myqueue.top().first);
            myqueue.pop();
        }
        return result;
    }
};

---

---

总结:
参考