2021-04-02 leetcode面试题 17.21. 直方图的水量

题目描述

给定一个直方图(也称柱状图)，假设有人从上面源源不断地倒水，最后直方图能存多少水量?直方图的宽度为 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）
链接：https://leetcode-cn.com/problems/volume-of-histogram-lcci

提交结果

解题思路

题目中是让求长方形柱子之间能容纳 的水量,首先需要 想清楚如何求 柱子之前水量
1.拿left 与max 之前来举例 ,因为 left 的高度是小的,所以能容纳的水应该 以left的高度来决定,只需要遍历从left 位置 到max 位置,用 left 的高度 减去 当前柱子的高度就是当前位置的水量,遍历求和就可以算出 left 到 right 之间的水量了, 同理可以求出 max 和right之间的.
2. 如何来找 max,left,right 呢, max 是最高的柱子,盛水 肯定需要先找到max,第一次遍历可以先找到 最高的柱子位置,也就是数组的索引 max 和柱子高度 height[max],然后再去找 max 左侧中最高的柱子 left,和max 右侧最高的柱子 right,这样就可以求出 left 到 right 之间的水量了,递归重复执行上述代码,将不同位置的水量都求出来.

实现代码

class Solution {
    public int trap(int[] height) {
        return getWater(0,height.length-1,height);
    }
    // 递归求数组 height中 start 到 end位置的水量 
    public int getWater(int start,int end,int[] height ){
        if(start>=end){
            return 0;
        }
        //求最高柱子位置
        int max = height[start];
        int maxI = start; 
        for(int i=start+1;i<=end;i++){
            if(height[i]>max){
                max = height[i];
                maxI = i;
            }
        }
        //求最高柱子左侧的最高柱子
        int left = height[start];
        int leftI = start; 
        for(int i=start+1;i<maxI;i++){
            if(height[i]>left){
                left = height[i];
                leftI = i;
            }
        }
        //求最高柱子右侧的最高柱子
        int right = height[end];
        int rightI = end; 
        for(int i=maxI+1;i<end;i++){
            if(height[i]>right){
                right = height[i];
                rightI = i;
            }
        }
        int sum = 0;
        //求left 与max 之间的水量
        for(int i=leftI+1;i<maxI;i++){
            sum += left-height[i];
        }
        //求max 与right 之间的水量
        for(int i=maxI+1;i<rightI;i++){
            sum += right-height[i];
        }
        //递归查询 left左侧的水量
        if(start<leftI){
            sum += getWater(start,leftI,height);
        }
        //递归查询right右侧的水量
        if(end>rightI){
            sum += getWater(rightI,end,height);
        }
        return sum; 
    }
}