[leetcode]42. 接雨水

42. 接雨水

给定 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
0 <= n <= 3 * 104
0 <= height[i] <= 105

动态规划：

func trap(height []int) int {
    res := 0
    n := len(height)
    if n < 3 {
        return res
    }

    leftMax := make([]int, n)
    leftMax[0] = height[0]
    for i := 1; i < n; i++ {
        leftMax[i] = max(leftMax[i-1], height[i])
    }
    rightMax := make([]int, n)
    rightMax[n-1] = height[n-1]
    for j := n-2; j >= 0; j-- {
        rightMax[j] = max(rightMax[j+1], height[j])
    }

    for i,h := range height {
        res += min(leftMax[i], rightMax[i]) - h
    } 
    return res
}

func min(a,b int) int {
    if a < b {
        return a
    }
    return b
}

func max(a,b int) int {
    if a < b {
        return b
    }
    return a
}

单调栈：

func trap(height []int) int {
    stack := []int{}
    res := 0
    for i,h := range height {
        for len(stack)>0 && h>height[stack[len(stack)-1]] {
            top := stack[len(stack)-1]
            stack = stack[:len(stack)-1]
            if len(stack) == 0 {
                break
            }
            left := stack[len(stack)-1]
            curWidth := i - left - 1
            curHeight := min(height[left], h) - height[top]
            res += curWidth * curHeight
        }
        stack = append(stack, i)
    }
    return res;
}

func min(a,b int) int {
    if a < b {
        return a
    }
    return b
}