代码随想录算法训练营第13天|239. 滑动窗口最大值,347. 前 K 个高频元素
239. 滑动窗口最大值
题目链接:239. 滑动窗口最大值,难度:困难
【实现代码】
class Solution {
private:
class MyQueue{
private:
deque<int> dq;
public:
MyQueue(){ }
void pop(int num){
if (!this->dq.empty() && this->dq.front() == num) {
this->dq.pop_front();
}
}
void push(int num) {
while (!this->dq.empty() && this->dq.back() < num) {
this->dq.pop_back();
}
this->dq.push_back(num);
}
int front() {
return this->dq.front();
}
};
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
vector<int> result;
MyQueue mq;
int index = 0;
for (int i = 0; i < k; i++) {
mq.push(nums[i]);
}
result.push_back(mq.front());
for (int i = k; i < nums.size(); i++) {
mq.pop(nums[i-k]);
mq.push(nums[i]);
result.push_back(mq.front());
}
return result;
}
};
【解题思路】
在解题之前先确定几个难点:
- 怎样计算窗口内的最大值
- 怎样保证单调队列是一个单调的状态
- 窗口移动时,怎样去掉单调队列中不在窗口内的最大值
解决方法:
- 创建一个单调队列,使用双端数组(deque )实现单调队列;
- 计算窗口内的最大值是,比较队列中的首元素与当前元素的值,小于当前值时,移除当前值的所有元素;
- 当入队列时,比较队尾元素与当前的值,小于当前值时就移除,直到大于当前值;
- 从第0个元素开始,如果在队列首元素等于当前索引的元素,则说明队列首元素不在窗口内了,移除它;
- 每移动一次窗口就记录队列的首元素(最大值)。
347. 前 K 个高频元素
题目链接:347. 前 K 个高频元素,难度:中等
【实现代码】
class Solution {
public:
class MyComparison{
public:
bool operator()(const pair<int, int>& p1, const pair<int, int>& p2) {
return p1.second > p2.second;
}
};
vector<int> topKFrequent(vector<int>& nums, int k) {
unordered_map<int, int> m;
for (int i = 0; i < nums.size(); i++) {
m[nums[i]]++;
}
priority_queue<pair<int, int>, vector<pair<int, int>>, MyComparison> pri_que;
for (unordered_map<int, int>::iterator it = m.begin(); it != m.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;
pri_que.pop();
}
return result;
}
};
【解题思路】
- 使用无序map存储值的个数(我想到了);
- 对map中值出现的个数进行排序,使用优先级队列(没想出来)
- 使用小顶堆存储前k个出现的值(第一次接触小顶堆)
- 再倒序存进vector(注意先申请空间)