239. 滑动窗口最大值
- 题目
给你一个整数数组 nums,有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k 个数字。滑动窗口每次只向右移动一位。返回 滑动窗口中的最大值 。
题目链接:https://leetcode.cn/problems/sliding-window-maximum/
- 代码
class Solution {
public:
//定义队列
class MyQueue{
public:
deque<int> que;
void pop(int val){
if(!que.empty() && val == que.front()) que.pop_front();
}
void push(int val){
while(!que.empty() && val > que.back()) que.pop_back();
que.push_back(val);
}
int getMax(){
return que.front();
}
};
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
MyQueue que;
vector<int> res;
for(int i = 0; i < k; i++) que.push(nums[i]);
res.push_back(que.getMax());
for(int i = 0; i < nums.size() - k ; i++){
que.pop(nums[i]);
que.push(nums[i + k]);
res.push_back(que.getMax());
}
return res;
}
};
- 小结
自定义一个单调队列,该队列使用双端队列为底层容器。定义队列的pop、push以及getMax操作。其中pop操作:根据窗口的移动而移出队列中非窗口内的元素,push操作:将队列中比窗口增加的元素小的元素从队尾移出直到队列中的全部元素都比需要增加的元素大,再从队尾压入该元素。
347.前 K 个高频元素
- 题目
给你一个整数数组 nums 和一个整数 k ,请你返回其中出现频率前 k 高的元素。你可以按 任意顺序 返回答案。
题目链接:https://leetcode.cn/problems/top-k-frequent-elements/
- 代码
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;//map<nums[i],对应出现的次数>
for(int i = 0; i < nums.size(); i++){
map[nums[i]]++;
}
//频率排序固定大小k,存放iterator
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){
pri_que.pop();
}
}
//输出前k个高频元素
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中的key来记录数组元素,value记录该元素的频率。定义大小为k的小顶堆,遍历map中的value,小顶堆每次加入新的元素时,需要移除内部频率最小的元素,保持小顶堆内始终为频率最大的k个元素。遍历小顶堆并输出频率最大的元素对应的整数。