本题要求滑动窗口区间内的最大值,需要利用单调队列的特性,使得队列从队头到队尾始终是递减排列,这样每次滑动窗口移动时弹出的队头元素即为滑动窗口区间内的最大值。此方法比暴力算法(每移动一次求一次窗口内最大值)时间复杂度低得多。
具体代码如下:
class Solution {
private:
class Myqueue{
public:deque<int>qu;
void pop(int val)
{
if(!qu.empty()&&val==qu.front())
{
qu.pop_front();
}
}
void push(int val)
{
while(!qu.empty()&&val>qu.back())
{
qu.pop_back();
}
qu.push_back(val);
}
int front()
{
return qu.front();
}
};
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
Myqueue qu;
vector<int>result;
for(int i=0;i<k;i++)
{
qu.push(nums[i]);
}
result.push_back(qu.front());
for(int i=k;i<nums.size();i++)
{
qu.pop(nums[i-k]);
qu.push(nums[i]);
result.push_back(qu.front());
}
return result;
}
};
本题要求数组内出现频率前k高的元素,考虑到要存储数组中的元素与元素出现的频率因此选用map存储结构,而后对map中的数据进行递增排序,将较小的数据优先弹出,直到剩下k个数据,这k个数据组成的数组即为所求。
具体代码如下:
class Solution {
public:
class compare{
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>num_map;
for(int i=0;i<nums.size();i++)
{
num_map[nums[i]]++;
}
priority_queue<pair<int,int>,vector<pair<int,int>>,compare>qu;
for(unordered_map<int,int>::iterator it=num_map.begin();it!=num_map.end();it++)
{
qu.push(*it);
if(qu.size()>k)
{
qu.pop();
}
}
vector<int>result;
for(int i=k-1;i>=0;i--)
{
result.push_back(qu.top().first);
qu.pop();
}
return result;
}
};