参考资料:
https://programmercarl.com/0239.%E6%BB%91%E5%8A%A8%E7%AA%97%E5%8F%A3%E6%9C%80%E5%A4%A7%E5%80%BC.html
https://programmercarl.com/0347.%E5%89%8DK%E4%B8%AA%E9%AB%98%E9%A2%91%E5%85%83%E7%B4%A0.html
题目描述:
给你一个整数数组 nums
,有一个大小为 k
的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k
个数字。滑动窗口每次只向右移动一位。
返回 滑动窗口中的最大值 。
示例 1:
输入:nums = [1,3,-1,-3,5,3,6,7], k = 3 输出:[3,3,5,5,6,7] 解释: 滑动窗口的位置 最大值 --------------- ----- [1 3 -1] -3 5 3 6 7 3 1 [3 -1 -3] 5 3 6 7 3 1 3 [-1 -3 5] 3 6 7 5 1 3 -1 [-3 5 3] 6 7 5 1 3 -1 -3 [5 3 6] 7 6 1 3 -1 -3 5 [3 6 7] 7
思路分析:
用队列维护当前窗口最大值。
push():value与队列前面的元素比较,将在value之前且小于value的元素弹走。
pop():移动窗口时,若头部为当前最大值,则弹走。(不是最大值的,在push就已经弹走了)
代码实现:
class Solution {
private:
class MyQueue {
public:
deque<int> que; //
void pop(int value) {//窗口移动时,若弹出的元素为当前的max
if (!que.empty() && value == que.front()) {//此处判断是否相等是为了避免 元素在之前的push已经被弹走(只有max才会被保留在队列头部)
que.pop_front();//从前面弹
}
}
void push(int value) {//将 在其之前加入且值小于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++) { //前k
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;
}
};
题目描述:
给你一个整数数组 nums
和一个整数 k
,请你返回其中出现频率前 k
高的元素。你可以按 任意顺序 返回答案。
示例 1:
输入: nums = [1,1,1,2,2,3], k = 2 输出: [1,2]
思路分析:
用小顶堆维护前k个频次最高的元素。(小顶堆每次弹出最小值)
1. 先用map存储(key,value),(数字,次数)
2. 建大小为k的小顶堆,依次遍历map。将元素依次插入小顶堆
3. 最后小顶堆倒序存储前k个频次最高的元素
代码实现:
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){
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;
}
};