239.滑动窗口最大值
这个题在单调栈那一系列的题也做过,但是做第二遍的时候还是有点懵哈,但是看了Carl视频发现,我的想法还停留在单调队列,没有进一步保持这个队列是绝对单调的,但是要想保持这个队列绝对单调,还需要考虑队列每个元素是在滑动窗口之内的,不能超过这个限制。我想到之前做单调队列的时候都会把下标存入数组或者队列,这样的话就可以保证每个元素是没有超过滑动窗口了。
class Solution {
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
deque<int> res;
vector<int> result(nums.size()-k+1,0);
for(int i=0;i<nums.size();i++) {
while(!res.empty()&&res.back()+k<=i) {
res.pop_back();
}
while(!res.empty()&&nums[res.front()]<nums[i]) {
res.pop_front();
}
res.push_front(i);
if(i>=k-1) {
result[i-k+1]=nums[res.back()];
}
}
return result;
}
};
第一个while就是要保持左端元素一定是k个之内的,第二个while就是保证这个队列绝对单调非递增的;
判断条件if,就是结果存入数组了。最主要的是,vector< int > result ,这个存结果的数组,最好是提前定好size,后面用push_back的话,会有性能上的一点弊端,因为vector扩大内存都是双倍(看编译器)扩的,最开始不定内存的话,后续就要加内存,还要重新开启新空间等。
carl代码随想录的写法更简洁,更易于理解。
347.前k个高频元素
之前做过一次,但是还是哈,对于C++的数据结构中大顶堆不怎么熟悉,本来是想用map的自排序功能,用multimap,然后频次作为key,按照key从大到小排序;当然也有sort,就是map转换为vector,然后排序。
写了一遍大顶堆,很省力。
学习一下大顶堆的如何自定义数据结构:
class Solution {
public:
struct cmp1{
public:
bool operator()(const pair<int,int> &A,const pair<int,int> &B) {
return A.second>B.second;
}
};
vector<int> topKFrequent(vector<int>& nums, int k) {
unordered_map<int,int> res;
for(int i=0;i<nums.size();i++) {
res[nums[i]]++;
}
priority_queue<pair<int,int>,vector<pair<int,int>>,cmp1> r;
for(auto e:res) {
r.push(e);
if(r.size()>k) {
r.pop();
}
}
vector<int> result(k,0);
for(int i=0;i<k;i++) {
result[i]=r.top().first;
r.pop();
}
return result;
}
};