题目
初见思路
239. 滑动窗口最大值
主要根据代码随想录上的思路来的,使用deque来实现一个我们自己的单调队列。
347.前 K 个高频元素
用map来保存出现的数字和对应的次数。用一个priority_queue (max_heap)来对数字的出现次数进行排序,pop掉前K个就行了。
更新思路
239. 滑动窗口最大值
流程:
- 将开头的k个值push存入窗口中(单调队列)。push的时候会将需要push的值与入口(back)处的值进行比较,如果需要push的更大那就移除入口处的那个值,以此保证front的值一直都是当前窗口最大的。储存最大值(front)。
- 开始移动窗口,pop掉窗口最左边的值(如果这个值是前一个窗口最大值的话,如果不是就没必要pop,之后push的时候会pop掉),push新的值(同样也要pop掉比较小的值),然后将front存入result。
- 重复第二步。
class MyQueue {
public:
deque<int> que;
void pop(int val){
if(!que.empty() && val == que.front()){
que.pop_front();
}
}
void push(int value){
while(!que.empty() && value > que.back()){
que.pop_back(); // remove smaller values
}
que.push_back(value);
}
int front(){
return que.front();
}
};
class Solution {
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
MyQueue q;
vector<int> res;
// first window
for(int i = 0; i < k; ++i){
q.push(nums[i]);
}
res.push_back(q.front());
for(int i = k; i < nums.size(); ++i){
q.pop(nums[i - k]); // pop the slided value
q.push(nums[i]); // push to back, pop smaller values
res.push_back(q.front()); // max val for this sliding window.
}
return res;
}
};
347.前 K 个高频元素
max heap父节点比子节点大,在max heap中储存一个pair,根据map中的次数进行排序。pop root添加进result中。
class Solution {
public:
vector<int> topKFrequent(vector<int>& nums, int k) {
unordered_map<int, int> map;
for(int n : nums){
++map[n];
}
vector<int> res;
// pair<first, second>: first is frequency, second is number
priority_queue<pair<int, int>> pq;
for(auto it = map.begin(); it != map.end(); it++){
pq.push(make_pair(it->second, it->first));
if(pq.size() > (int)map.size() - k){ //k frequent
res.push_back(pq.top().second);
pq.pop();
}
}
return res;
}
};
今日总结
学习时长
2h
收获
https://programmercarl.com/%E6%A0%88%E4%B8%8E%E9%98%9F%E5%88%97%E6%80%BB%E7%BB%93.html