代码随想录算法训练营第十三天|LeetCode239 滑动窗口最大值、LeetCode347 前k个高频元素

题1：

指路：LeetCode239 滑动窗口最大值

思路与代码：

自定义的队列实现pop()，push()，getMaxvalue()三个函数，使其能一直维护有可能的最大值。好难，我还没完全会。代码如下：

class Solution {
    private:
    class MyQueue {
        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) {
    MyQueue que;
    vector<int> result;
    for (int i = 0; i < k; i++) {
        que.push(nums[i]);
    }
    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;
    }
};

题2：

指路：LeetCode347 前k个高频元素

思路与代码：

1.map

第一反应是用map键值对分别存放元素位置及出现次数再进行排序，最后输出前k个。但我的STL水平还没有这么高，只能借鉴一下大佬的。代码如下：

class Solution {
public:
    vector<int> topKFrequent(vector<int>& nums, int k) {
    map<int, int> map1;  // 键为元素，值为出现的次数
    for (int& n : nums ) {
       map1[n]++;
    }
    vector<pair<int, int>> nVec(map1.begin(), map1.end()); 
    // 以元素出现的次数为基准进行从小到大的排序
    sort(nVec.begin(), nVec.end(), [&](const pair<int, int> p1, const pair<int, int> p2) {
        return p1.second > p2.second;
    });
    vector<int> ans(k);
    transform(nVec.begin(), nVec.begin() + k, ans.begin(), [](const auto& p) {
        return p.first;
    });
    return ans;
    }
};

2.小顶堆

我们用小顶堆遍历map，在堆内部始终维护k个元素，push一个数，pop一个较小数。代码如下：

class Solution {
public:

    class mycomparison {
        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>>, mycomparison> pri_que;
  for (unordered_map<int, int>::iterator it = map.begin(); it != map.end(); it++) {
        pri_que.push(*it);
            if (pri_que.size() > k) { // 如果堆的大小大于了K，则队列弹出，保证堆的大小一直为k
                pri_que.pop();
            }
        }
    vector<int> result(k);
    for (int i = k - 1; i >= 0; i--) {
        result[i] = pri_que.top().first;
        pri_que.pop();
    }
    return result;
    }
};