在做这道题时,首先想到的解法是使用队列来做,维护一个队列,每次保存滑动窗口大小的长度,并判断此时队列中的最大值,但这样做,在k的值较大时,出现了超时问题,代码如下:
class Solution { public: vectormaxSlidingWindow(vector & nums, int k) { vector result; queue que; for(int i = 0; i < nums.size(); i++){ que.push(nums[i]); while(que.size() == k){ int maxNum = que.front(); que.pop(); que.push(maxNum); for(int flag = 0; flag < k - 1; flag++){ int tmp = que.front(); maxNum = max(maxNum, tmp); que.pop(); que.push(tmp); } result.push_back(maxNum); que.pop(); } } return result; } };
使用deque双端队列来完成这道题,首先遍历前k个元素,将最大值的下标加入到队列中,如果新加入的下标对应的值大于队列前面下标对应的值,将其移除。这样保持这个队列维护的下标,对应的值时由大到小单调的。
之后每新插一个元素进来,继续维护这个单调的队列,然后判断队列最前的下标,是否还在滑动窗口内,如果不在,则移除。代码如下:
class Solution { public: vectormaxSlidingWindow(vector & nums, int k) { vector result; deque dq; for(int i = 0; i < k; i++){ while(!dq.empty() && nums[i] >= nums[dq.back()]){ dq.pop_back(); } dq.push_back(i); } result.push_back(nums[dq.front()]); for(int i = k; i < nums.size(); i++){ while(!dq.empty() && nums[i] >= nums[dq.back()]){ dq.pop_back(); } dq.push_back(i); while(dq.front() <= i - k){ dq.pop_front(); } result.push_back(nums[dq.front()]); } return result; } };
还没有评论,来说两句吧...