Leetcode42.接雨水

题目描述

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

示例：

上面是由数组 [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

题解

如下多个题解摘自于【Leetcode每日打卡】接雨水问题的超完全手册。🐂🍺。

暴力法（java）

第一个方法的思路很清晰，计算图中每一个矩形上方水（蓝色区域）的面积，遍历每一个矩形（即数组中的每一个元素），然后得到面积和。计算面积的方式就是把该柱子的左最高和右最高找到，取较小值再减去该柱子的高度。

class Solution {
    public int trap(int[] height) {
        int res = 0;
        // 遍历每个柱子
        // 因为第零个和最后一个柱子上肯定没水，故直接跳过
        for (int i = 1; i < height.length - 1; i++) {
            int leftMax = 0, rightMax = 0;
            // 计算当前柱子左侧的柱子中的最大高度
            for (int j = 0; j <= i; j++) {
                leftMax = Math.max(leftMax, height[j]);
            }
            // 计算当前柱子右侧的柱子中的最大高度
            for (int j = i; j < height.length; j++) {
                rightMax = Math.max(rightMax, height[j]);
            }
            // 结果中累加当前柱子顶部可以储水的高度，
            // 即 当前柱子左右两边最大高度的较小者 - 当前柱子的高度。
            res += Math.min(leftMax, rightMax) - height[i];
        }
        return res;
    }
}

复杂度分析

• 时间复杂度：$O(n^2)$
• 空间复杂度：$O(1)$

单调栈（java）

单调栈即单调递增或单调减的栈。维护单调栈，即如果新加元素会使栈不满足单调性，则需pop出相应元素使得新加入元素后还保持单调性。
对于本题，如果新入栈元素小于等于栈顶元素，则无法形成凹槽，故入栈；如果大于新入栈元素大于栈顶元素，则会产生凹槽。同时对于栈来说，不满足单调性，需要将栈中小于当前柱子的元素pop出来，将凹槽的大小累加到结果中。

public class Solution {
    public int trap(int[] height) {
        if (height == null) {
            return 0;
        }
        Stack<Integer> stack = new Stack<>();
        int ans = 0;
        for (int i = 0; i < height.length; i++) {
            while(!stack.isEmpty() && height[stack.peek()] < height[i]) {
                int curIdx = stack.pop();
                // 如果栈顶元素一直相等，那么全都pop出去，只留第一个。
                while (!stack.isEmpty() && height[stack.peek()] == height[curIdx]) {
                    stack.pop();
                }
                if (!stack.isEmpty()) {
                    int stackTop = stack.peek();
                    // stackTop此时指向的是此次接住的雨水的左边界的位置。右边界是当前的柱体，即i。
                    // Math.min(height[stackTop], height[i]) 是左右柱子高度的min，减去height[curIdx]就是雨水的高度。
                    // i - stackTop - 1 是雨水的宽度。
                    ans += (Math.min(height[stackTop], height[i]) - height[curIdx]) * (i - stackTop - 1);
                }
            }
            stack.add(i);
        }
        return ans;
    }
}

• 时间复杂度：O(n)
• 空间复杂度：O(n)

动态规划（java）

降维，空间换时间。思路同暴力法， 但是可以先构建一个dp数组，保存每一个柱子左右侧的最大值，然后再挨个柱子计算其蓝色区域面积。

class Solution {
    public int trap(int[] height) {
        int n = height.length;
        if (n == 0) {
            return 0;
        }
        // 定义二维dp数组
        // dp[i][0] 表示下标i的柱子左边的最大值
        // dp[i][1] 表示下标i的柱子右边的最大值
        // 构建（n x 2）的数组来更新最大值
        int[][] dp = new int[n][2];
        //初始化
        dp[0][0] = height[0];
        dp[n - 1][1] = height[n - 1];
        //更新
        for (int i = 1; i < n; i++) {
            dp[i][0] = Math.max(height[i], dp[i - 1][0]);   
        }
        for (int i = n - 2; i >= 0; i--) {
            dp[i][1] = Math.max(height[i], dp[i + 1][1]);
        }
        // 遍历每个柱子，累加当前柱子顶部可以储水的高度，
        // 即 当前柱子左右两边最大高度的较小者 - 当前柱子的高度。
        int res = 0;
        for (int i = 1; i < n - 1; i++) {
            res += Math.min(dp[i][0], dp[i][1]) - height[i];
        } 
        return res;
    }
}

• 时间复杂度：O(n)
• 空间复杂度：O(n)

2020.4.5