数据结构与算法 - 单调队列和栈

一、单调递减队列

package com.itheima.algorithms.leetcode;

import java.util.LinkedList;

public class MonotonicQueue {

    private LinkedList<Integer> deque = new LinkedList<>();

    public Integer peek() {
        return deque.peekFirst();
    }

    public void poll() {
        deque.pollFirst();
    }

    public void offer(Integer t) {
        while(!deque.isEmpty() && deque.peekLast() < t) {
            deque.pollLast();
        }
        deque.offerLast(t);
    }

    @Override
    public String toString() {
        return deque.toString();
    }

    public static void main(String[] args) {
        MonotonicQueue q = new MonotonicQueue();
        for (int i : new int[]{1, 3, -1, -3, 5, 3, 6, 7}) {
            q.offer(i);
            System.out.println(q);
        }
    }
}

运行结果：

[1]
[3]
[3, -1]
[3, -1, -3]
[5]
[5, 3]
[6]
[7]

1.1 最大滑动窗口

给你一个整数数组 nums，有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k 个数字。滑动窗口每次只向右移动一位。

返回 滑动窗口中的最大值 。

示例 1：

输入：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

示例 2：

输入：nums = [1], k = 1
输出：[1]

提示：

• 1 <= nums.length <= 10^5
• -10^4 <= nums[i] <= 10^4
• 1 <= k <= nums.leng

class Solution {
    static class MonotonicQueue {

        private LinkedList<Integer> deque = new LinkedList<>();

        public Integer peek() {
            return deque.peekFirst();
        }

        public void poll() {
            deque.pollFirst();
        }

        public void offer(Integer t) {
            while (!deque.isEmpty() && deque.peekLast() < t) {
                deque.pollLast();
            }
            deque.offerLast(t);
        }

        @Override
        public String toString() {
            return deque.toString();
        }
    }

    public int[] maxSlidingWindow(int[] nums, int k) {
        MonotonicQueue q = new MonotonicQueue();
        int[] result = new int[nums.length - (k - 1)];
        for (int i = 0; i < nums.length; i++) {
            // 如果队列头部的元素超过了滑动窗口的范围，则移除
            if (i >= k && nums[i - k] == q.peek()) {
                q.poll();
            }
            int num = nums[i];
            q.offer(num);
            // 索引大于等于k时才开始计算滑动窗口内的最大值
            if (i >= k - 1) {
                // 单调递减队列队头的元素是最大的
                result[i - (k - 1)] = q.peek();
            }
        }

        return result;
    }
}

二、单调递减栈

2.1 接雨水

给定 n 个非负整数表示每个宽度为 1 的柱子的高度图，计算按此排列的柱子，下雨之后能接多少雨水。

示例 1：

​

输入：height = [0,1,0,2,1,0,1,3,2,1,2,1]
输出：6
解释：上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图，在这种情况下，可以接 6 个单位的雨水（蓝色部分表示雨水）。 

示例 2：

输入：height = [4,2,0,3,2,5]
输出：9

提示：

• n == height.length
• 1 <= n <= 2 * 10^4
• 0 <= height[i] <= 10^5

class Solution {
    static class Data {
        int height;
        int i; // 索引

        public Data(int height, int i) {
            this.height = height;
            this.i = i;
        }

        @Override
        public String toString() {
            return String.valueOf(height);
        }
    }

    /**
     * 1. 维护一个单调递减栈
     * 2. 当加入一个新元素时，如果发现需要弹出元素，表示遇到了一个凹陷的位置，此时应该计算雨水容量
     * 
     * @param height
     * @return
     */
    public int trap(int[] height) {
        LinkedList<Data> stack = new LinkedList<>();
        int total = 0;
        for (int i = 0; i < height.length; i++) {
            Data right = new Data(height[i], i);

            while (!stack.isEmpty() && stack.peek().height < right.height) {
                Data pop = stack.pop();
                Data left = stack.peek();
                if (left != null) { // 计算水的容量
                    int width = right.i - left.i - 1;
                    int h = Math.min(left.height, right.height) - pop.height;
                    total += width * h;
                }
            }
            stack.push(right);
        }

        return total;
    }
}