代码随想录算法训练营第十三天|239. 滑动窗口最大值、347.前 K 个高频元素
本天的题解思路代码均来自于y总的:http://www.acwing.com
239. 滑动窗口最大值
我是一看就会,一写就费,先看代码:
class Solution {
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
vector<int> q(nums.size());
vector<int> res;
int hh = 0,tt = -1;
for(int i = 0;i < nums.size();i++)
{
if(hh <= tt && i - k + 1 > q[hh]) hh++;
while(hh <= tt && nums[i] > nums[q[tt]]) tt--;
q[++tt] = i;
if(i >= k - 1) res.push_back(nums[q[hh]]);
}
return res;
}
};
这道题目的主要就是用到了单调队列。为什么叫单调队列呢,因为当一系列操作之后,队列里面保留的数值(实际上队列里存的是数值的下标)是单调的,这里还有一个概念就是滑动窗口,这个概念比较好理解,就是一个定长的窗口在字符串上滑动,然后选择出窗口中的最小值或者最大值。我的代码和y总的几乎一样,不过我用的他基础课里面的方法,就是自己用数组构造一个队列。
这道题目的总体思路其实不难,难的是代码逻辑,所以我想说一下代码逻辑,首先创建一个数组当作队列,然后创建每次滑动窗口保留的数值的数组,还有两个指针,一个是hh,表示队头,另一个是tt表示队尾,这里的队列不是普通队列,而是双端队列,双端队列的特点就是,两端都能进队,两端都能出队。hh是对头,tt是队尾,但是赋值-1就代表目前队列里面还没有元素,然后遍历每个数,首先第一个if语句,就是说如果队列里面有元素,而且i-k+1(代表滑动窗口的第一个窗口)>q[hh],说明滑动窗口的最小的位置比队列的头部存储的下表还大,说明滑动窗口移动了,所hh该往前走了,最关键的一句来了,while语句,条件是只要队列不为空,而且后面来的这个值比队列里存储的前一个位置的数都大,那么前一个数只要的在后面出现的滑动窗口中就别想是最大的了,所以,长江后浪推前浪,前浪被排出队列,同理,前面的前面如果也小,也一个下场,最后留下最大的,然后,然后q[++tt] = i;开始补数,有人会问,如果后面来的数比前面队列位置里面的小怎么办,那前面队列就不会被排出去,因为后面来了个小卡拉米,自己在滑动窗口的推移期间还是有机会当上大哥的。最后如果i >= k - 1;也就是说,如果这个滑动窗口是满的话才能输出,比如刚开始,滑动窗口是由空闲的,那就不能输出。
347.前 K 个高频元素
先看代码:
class Solution {
public:
vector<int> topKFrequent(vector<int>& nums, int k) {
unordered_map<int,int> cnt;
for(auto num:nums) cnt[num]++;
int n = nums.size();
vector<int> s(n + 1);
for(auto [x,c]:cnt) s[c]++;
vector<int> res;
int i = n,t = 0;
while(t < k) t += s[i--];
for(auto [x,c]:cnt)
{
if(c > i) res.push_back(x);
}
return res;
}
};