1、LeetCode239
LeetCode239
class Solution {
private:
class MyDeque {
public:
deque<int> que;
void pop(int value) {
if (!que.empty() && value == que.front()) {
que.pop_front();
}
}
// 将元素加入队列,元素大于队尾的元素,则弹出队尾元素,直至元素小于队尾元素或队空
void push(int 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) {
MyDeque que;
vector<int> result;
// 先处理前k个元素
for (int i = 0; i < k; i++) {
que.push(nums[i]);
}
// 将前k个元素中最大元素(即队头元素)的push进result中
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()); // 将前k个元素中最大元素(即队头元素)的push进result中
}
return result;
}
};
2、LeetCode347
LeetCode347
class Solution {
public:
class mycomparssion {
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;
// 先用哈希map映射,统计元素次数
for (int i = 0; i < nums.size(); i++) {
map[nums[i]]++;
}
// 建立优先队列
priority_queue<pair<int, int>, vector<pair<int, int>>, mycomparssion> 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; // top()为 pair<int, int>类型
pri_que.pop();
}
return result;
}
};