LeetCode:239. Sliding Window Maximum

写这一题是因为这一题有特定的解法,记录一下防止忘记。
1.特定解法
针对这一题有一种专门的数据结构,叫做Monotonic Queue。一个deque,里面存放的是数组的索引,从前往后代表了下一个最大值的候选者。每次迭代到一个新的数组元素:1)看看需不需要淘汰最大的那个索引,因为它可能随着窗口的滑动而到外面去了。2)循环的将deque的尾元素代表的索引值对应的数组中的值和当前的那个值比较,如果当前的那个值大,就把尾元素弹出,再接着比较下一个尾元素,直到遇到比它大的或者将全部元素都弹出,再将当前的索引压入。3)因为deque的首元素代表的索引值对应的数组中的值是当前窗口中最大的,所以将对应元素压入返回的vector中。
其实直接看代码理解更简单
class Solution {
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
deque<int> dq;
vector<int> res;
for (int i = 0; i < nums.size(); ++i) {
if (!dq.empty() && dq.front() <= i-k) dq.pop_front();
while (!dq.empty() && nums[dq.back()] <= nums[i])//这里小于或者小于等于都行
dq.pop_back();
dq.push_back(i);
if (i >= k-1) res.push_back(nums[dq.front()]);//在形成窗口之后才压入最大值
}
return res;
}
};
2.我自己想的一个方法
还算做了一定的优化,不是那种暴力的解法,仅供参考
class Solution {
public:
vector<int> maxSlidingWindow(vector<int>& nums, int k) {
vector<int> ret;
if (nums.empty() || k > nums.size())
return ret;
int maxIndex = findMax(nums, 0, k);
int maxVal = nums[maxIndex];
ret.push_back(maxVal);
int remain = maxIndex - 0;
for (int i = k; i < nums.size(); ++i) {
if (nums[i] >= maxVal) {
maxVal = nums[i];
remain = k-1;
ret.push_back(maxVal);
}
else {
--remain;
if (remain < 0) {
maxIndex = findMax(nums, i-k+1, k);
maxVal = nums[maxIndex];
remain = maxIndex - (i-k+1);
ret.push_back(maxVal);
}
else ret.push_back(maxVal);
}
}
return ret;
}
private:
int findMax(const vector<int> &nums, int index, int k) {
int maxVal = INT_MIN;
int ret = index;
for (int i = index; i < index+k; ++i) {
if (nums[i] > maxVal) {
maxVal = nums[i];
ret = i;
}
}
return ret;
}
};
浙公网安备 33010602011771号