代码随想录算法13 | LC239、347

Leetcode239. 滑动窗口最大值

题目链接：

[239. 滑动窗口最大值 - 力扣（Leetcode）]

思路：

设计单调队列的时候，pop，和push操作要保持如下规则：

1. pop(value)：如果窗口移除的元素value等于单调队列的出口元素，那么队列弹出元素，否则不用任何操作
2. push(value)：如果push的元素value大于入口元素的数值，那么就将队列入口的元素弹出，直到push元素的数值小于等于队列入口元素的数值为止

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

Leetcode347. 前 K 个高频元素

题目链接：

[347. 前 K 个高频元素 - 力扣（Leetcode）]

思路：

我们要用小顶堆，因为要统计最大前k个元素，只有小顶堆每次将最小的元素弹出，最后小顶堆里积累的才是前k个最大元素。

即优先级队列

class Solution {
public:
    class com{
        public:
            bool operator()(const pair<int,int> &l, const pair<int,int> &r)
            {
                return l.second>r.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>>, com> pri_que;
        for(auto it=map.begin();it!=map.end();it++)
        {
            pri_que.push(*it);
            if(pri_que.size()>k)
            {
                pri_que.pop();
            }
        }
        vector<int> res(k);
        for(int i=k-1;i>=0;i--)
        {
            res[i]=pri_que.top().first;
            pri_que.pop();
        }
        return res;

    }
};

总结

1. priority_queue<int, vector<int>, greater<int>>即声明小根堆，类比代码自定义的运算比较规则
2. *it对迭代器取指即为本来的值
3. vector < int > v(n);初始化类似于resize

今天多次运用迭代器知识，十分不熟悉，很有收获