13代码随想录训练营day13|栈与队列part03

1、LeetCode239

LeetCode239

class Solution {
private:
    class MyDeque {
        public:
            deque<int> que;
            void pop(int value) {
                if (!que.empty() && value == que.front()) {
                    que.pop_front();
                }
            }
			// 将元素加入队列，元素大于队尾的元素，则弹出队尾元素，直至元素小于队尾元素或队空
            void push(int value) {
                while (!que.empty() && value > que.back()) {
                    que.pop_back();
                }
                que.push_back(value);
            }
			// 获取队头元素
            int front() {
                return que.front();
            }
    };
public:
    vector<int> maxSlidingWindow(vector<int>& nums, int k) {
        MyDeque que;
        vector<int> result;
        // 先处理前k个元素
        for (int i = 0; i < k; i++) {
            que.push(nums[i]);
        }
        // 将前k个元素中最大元素（即队头元素）的push进result中
        result.push_back(que.front());
        for (int i = k; i < nums.size(); i++) {
            que.pop(nums[i - k]);		// 弹出移动窗口后，移除出窗口中的元素（前提是该元素为最大元素）
            que.push(nums[i]);
            result.push_back(que.front());	// 将前k个元素中最大元素（即队头元素）的push进result中
        }
        return result;
    }
};

2、LeetCode347

LeetCode347

class Solution {
public:
    class mycomparssion {
    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;
        // 先用哈希map映射，统计元素次数
        for (int i = 0; i < nums.size(); i++) {
            map[nums[i]]++;
        }
        // 建立优先队列
        priority_queue<pair<int, int>, vector<pair<int, int>>, mycomparssion> pri_que;
        // 迭代器遍历
        for (unordered_map<int, int>::iterator it = map.begin(); it != map.end(); it++) {
            pri_que.push(*it);
            if (pri_que.size() > k) {
                pri_que.pop();
            }
        }
        vector<int> result(k);
        for (int i = k - 1; i >= 0; i--) {
            result[i] = pri_que.top().first;	// top()为 pair<int, int>类型
            pri_que.pop();
        }
        return result;
    }
};