剑指Offer 59 I II

179 阅读2分钟

这是我参与8月更文挑战的第31天,活动详情查看:8月更文挑战

剑指 Offer 59 - I. 滑动窗口的最大值

题目

给定一个数组 nums 和滑动窗口的大小 k,请找出所有滑动窗口里的最大值。

示例:

输入: nums = [1,3,-1,-3,5,3,6,7], 和 k = 3
输出: [3,3,5,5,6,7] 
解释: 
 滑动窗口的位置                最大值
---------------               -----
[1  3  -1] -3  5  3  6  7       3
 1 [3  -1  -3] 5  3  6  7       3
 1  3 [-1  -3  5] 3  6  7       5
 1  3  -1 [-3  5  3] 6  7       5
 1  3  -1  -3 [5  3  6] 7       6
 1  3  -1  -3  5 [3  6  7]      7

提示:

你可以假设 k 总是有效的,在输入数组不为空的情况下,1 ≤ k ≤ 输入数组的大小。

方法一

滑动窗口模板:维护一个队列,队列中存放当前窗口中元素的值的,对头为窗口中的最大值;每次向后滑动时,判断对头元素是否已经滑出窗口外;每次要向队列中加入元素时,弹出队尾比当前要加入元素小的元素;

class Solution {
    public int[] maxSlidingWindow(int[] nums, int k) {
        int n = nums.length, idx = 0;
        if (n == 0) return new int[0];
        int[] res = new int[n - k + 1];
        LinkedList<Integer> q = new LinkedList<>();
        for (int i = 0; i < n; i ++ ) {
            if (q.size() > 0 && i - k >= q.getFirst()) q.poll();
​
            while(q.size() > 0 && nums[i] >= nums[q.getLast()]) q.removeLast();
            q.addLast(i);
            if (i >= k - 1) res[idx ++] = nums[q.getFirst()];
        }
        return res;
​
    }
}

时间复杂度: O(n)

空间复杂度: O(k)

剑指 Offer 59 - II. 队列的最大值

题目

请定义一个队列并实现函数 max_value 得到队列里的最大值,要求函数max_valuepush_backpop_front均摊时间复杂度都是O(1)。

若队列为空,pop_frontmax_value 需要返回 -1

示例 1:

输入:
["MaxQueue","push_back","push_back","max_value","pop_front","max_value"]
[[],[1],[2],[],[],[]]
输出: [null,null,null,2,1,2]

示例 2:

输入: 
["MaxQueue","pop_front","max_value"]
[[],[],[]]
输出: [null,-1,-1]

限制:

  • 1 <= push_back,pop_front,max_value的总操作数 <= 10000
  • 1 <= value <= 10^5

方法一

维护一个双端队列d,来记录队列中的最大值

  • 每次加入一个元素时,弹出d中小于新加入这个值的尾部元素
  • 删除一个元素时,判断d中的对头的元素是不是也等于这元素,如果是,也要一并删除;
class MaxQueue {
    LinkedList<Integer> q1, q2;
    public MaxQueue() {
        q1 = new LinkedList<>();
        q2 = new LinkedList<>();
    }
    
    public int max_value() {
        if (q2.size() == 0) return -1;
        return q2.getFirst();
    }
    
    public void push_back(int value) {
        q1.add(value);
        while(q2.size() > 0 && q2.getLast() < value) q2.removeLast();
        q2.add(value);
    }
    
    public int pop_front() {
        if (q1.size() == 0) return -1;
        int t = q1.poll();
        if (q2.getFirst() == t) q2.poll();
        return t;
    }
}
​
/**
 * Your MaxQueue object will be instantiated and called as such:
 * MaxQueue obj = new MaxQueue();
 * int param_1 = obj.max_value();
 * obj.push_back(value);
 * int param_3 = obj.pop_front();
 */