LeetCode 239 滑动窗口最大值 2023.11.6
- 题目链接
- 代码随想录讲解[链接]
- 题意:输入一个整数数组,然后用一个大小为k的窗口依次从数组开始滑向末尾,输出每个窗口内的最大值保存到一个数组中,最后输出该数组
- 方法:对窗口内的数据结构及接口自定义操作,例如窗口滑动时,删除窗口前一个元素需判断其是否为窗口的最大值,添加窗口尾元素时,先将窗口内比该值小的都删除,最后放入该元素(排序),窗口滑动时记录最大值到返回的数组中
class Solution {
public:
//对删除操作进行定义,删除需判断队列是否为空且要删除的元素是否为该窗口的最大值
void pop(deque<int> &dq, int val)
{
if(!dq.empty() && val == dq.front())
dq.pop_front();
}
//对添加操作进行定义,对窗口内元素进行排序,当窗口内元素小于要加入元素时删除,最后加入该元素
void push(deque<int> &dq, int val)
{
while (!dq.empty() && dq.back() < val)
dq.pop_back();
dq.push_back(val);
}
//获得最大值,即返回队列首的元素
int get_max(deque<int> &dq)
{
if(!dq.empty())
return dq.front();
return {};
}
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
deque<int> dq;
vector<int> result;
for(int i = 0; i < nums.size(); i++)
{
//当窗口还没“建好”时,只添加,不删除
if(i < k-1)
push(dq, nums[i]);
//当窗口刚“建好”时,添加后记录最大值
else if (i == k-1)
{
push(dq, nums[i]);
result.push_back(get_max(dq));
}
//当窗口移动时,先删除再添加后记录
else
{
pop(dq, nums[i-k]);
push(dq, nums[i]);
result.push_back(get_max(dq));
}
}
return result;
}
};
LeetCode 347 前k个高频元素 2023.11.6
-
大小顶堆都类似一个滑动窗口,窗口滑动时,大顶堆是出一个大元素,进一个小元素;小顶堆是出一个小元素,进一个大元素
-
题意:输入一个整数数组,输出其中出现频率前k高的元素的一个数组
-
方法:先用map键值对存储数组中出现元素及其出现次数
- 用大小堆优先队列(滑动窗口)记录频率前k的元素
- 或对map键值对中的值进行排序,记录前k的元素,参考文章[C++中的map排序链接]
//用哈希表排序的方法
//仿函数,参与vector降序排序
bool cmp(const pair<int, int> a, pair<int, int> b)
{
return a.second > b.second;
}
class Solution {
public:
vector<int> topKFrequent(vector<int>& nums, int k) {
//创建存放答案的数组
vector<int> result;
//因为题目数组长度可能较大,所以用无序map存储每个元素出现的次数
unordered_map<int, int> unmap;
for (int i = 0; i < nums.size(); i++)
unmap[nums[i]]++;
//测试发现不能对无序map的值进行排序操作,因此选用数组pair数据结构类型进行降序
//拷贝构造数组
vector<pair<int, int>> v(unmap.begin(), unmap.end());
//对数组的键值对中的值进行降序排序
sort(v.begin(), v.end(), cmp);
//将出现次数前k的键值对中的键进行记录
for (int i = 0; i < k; i++)
result.push_back(v[i].first);
return result;
}
};
//大小堆的方法
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) {
//创建存放答案的数组
vector<int> result;
//因为题目数组长度可能较大,所以用无序map存储每个元素出现的次数
unordered_map<int, int> unmap;
for (int i = 0; i < nums.size(); i++)
unmap[nums[i]]++;
// 对频率排序
// 定义一个小顶堆,大小为k
priority_queue<pair<int, int>, vector<pair<int, int>>, mycomparison> pri_que;
// 用固定大小为k的小顶堆,扫面所有频率的数值
for (unordered_map<int, int>::iterator it = unmap.begin(); it != unmap.end(); it++) {
pri_que.push(*it);
if (pri_que.size() > k) { // 如果堆的大小大于了K,则队列弹出,保证堆的大小一直为k
pri_que.pop();
}
}
// 找出前K个高频元素,因为小顶堆先弹出的是最小的,所以倒序来输出到数组
for (int i = k - 1; i >= 0; i--) {
result[i] = pri_que.top().first;
pri_que.pop();
}
return result;
}
};