-面试题 17.21. 直方图的水量

题目描述

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

样例

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

思路

这个题主要是要知道，每个位置 i 的储水后最大高度是多少？
很容易看出，每个位置 i 储水后最大高度取决于 i 两侧（不一定就是相邻的两个）柱子的最大高度中较小的那一个。就像一个“凹”下去的块，比如柱子高度为[3,2,0,1,4]，那么中间3个位置储水后都可以达到 “3”的高度。

只要知道每个位置 i 左右两侧柱子的最大高度，再减去位置 i 柱子的高度，就是位置 i 的储水量。

定义两个数组left和right。
left[i] 表示从 0 到 i 最大的柱子高度。left[i] = max(left[i-1],height[i])，left[0] = height[0]。从左往右遍历。
right[i] 表示从len - 1到 i 最大的柱子高度。right[i] = max(right[i+1],height[i])，right[len-1] = height[len-1]。从右往左遍历。

得到这两个数组后，我们只要遍历一遍height，然后求出每个 min(left[i], right[i]) - height[i]，把这个值累加起来就行。

注意，当 height 的 len <= 2时，直接return 0。

代码

class Solution {
    public int trap(int[] height) {
    	int len = height.length;
    	if(len <= 2)
    		return 0;
    		
    	int left[] = new int[len];
    	int right[] = new int [len];
    	int total = 0;
    	left[0] = height[0];
    	right[len-1] = height[len-1];
    	for(int i = 1;i < len;i++) {
    		left[i] = Math.max(left[i-1],height[i]);
    	}
    	for(int i = len - 2;i >= 0;i--) {
    		right[i] = Math.max(right[i+1], height[i]);
    	}
    	for(int i = 0;i < len;i++) {
    		int tmp = Math.min(left[i], right[i]);
    		total += (tmp - height[i]);
    	}
    	return total;
    }
}