面试题 17.21. 直方图的水量

题目

给定一个直方图(也称柱状图)，假设有人从上面源源不断地倒水，最后直方图能存多少水量?直方图的宽度为 1。

示例：

输入: [0,1,0,2,1,0,1,3,2,1,2,1]
输出: 6

分析： 只有凹槽处能存水，判断是否为凹槽只需要判断左右两边是否均存在高度大于自己的直方图，且该处存水的量取决于左右两边直方图高度较小的那个与自身的高度差，因此，一种暴力解法是：遍历每一个位置，对当前位置找到左侧大于自身高度的直方图，右侧大于自身高度的直方图，然后求较小的那个与自身的高度差，即为当前位置可存水的量。第二种解法是动态规划，由于当前位置存水量取决于左侧最高直方图和右侧最高直方图，因此可以创建两个数组，一个存放左侧最高直方图的高度，一个存放右侧最高直方图的高度，最后取二者的较小值与当前位置的高度做差，即为存水量

方法1： 遍历，暴力解法，时间复杂度为$O(n^2)$

class Solution:
    def trap(self, height: List[int]) -> int:
        count = 0
        for i in range(1, len(height)-1):
            h = height[i]
            left_max = h
            right_max = h
            for j in range(i):
                if height[j] > left_max:
                    left_max = height[j]
            for k in range(i+1, len(height)):
                if height[k] > right_max:
                    right_max = height[k]
            if left_max > h and right_max > h:
                count += (min(left_max, right_max) - h)
        return count

方法2： 动态规划，分别创建两个数组用来存放左侧和右侧的最高值，时间复杂度为$O(n)$, 空间复杂度为$O(n)$

class Solution:
    def trap(self, height: List[int]) -> int:
        left_height = [0]*len(height)
        right_height = [0]*len(height)
        if not height:
            return 0
        n = len(height)
        left_height[0] = height[0]
        right_height[-1] = height[-1]
        for i in range(1, n):
            left_height[i] = max(left_height[i-1], height[i])
            right_height[n-i-1] = max(right_height[n-i], height[n-i-1])
        h_sum = 0
        for i in range(n):
            h_sum += (min(left_height[i], right_height[i]) - height[i])
        return h_sum