LintCode:M-接雨水

LintCode链接

Given n non-negative integers representing an elevation map where the width of each bar is 1, compute how much water it is able to trap after raining.

LintCode:M-接雨水_第1张图片

样例

如上图所示,海拔分别为 [0,1,0,2,1,0,1,3,2,1,2,1], 返回 6.


(1)找到最高高度

(2)左右两边分别向最高处逼近,计算第一个比当前高度高的位置距离,以及中间覆盖的低于当前高度的高度总和,用当前高度能盛的最大和减去内部高度和,等于雨水和

public class Solution {
    /*
     * @param heights: a list of integers
     * @return: a integer
     */
    public int trapRainWater(int[] heights) {
        int n = heights.length;
        int max=0;
        int maxIndex=0;
        for(int i=0; iheights[j]){
                inner+=heights[j++];
            }
            vol+=heights[i]*(j-i-1)-inner;
            i=j++;
        }
        
        i=n-1; j=n-2;
        while(j>=maxIndex){
            int inner=0;
            while(heights[i]>heights[j]){
                inner+=heights[j--];
            }
            vol+=heights[i]*(i-j-1)-inner;
            i=j--;
        }
        
        return vol;
    }
}


你可能感兴趣的:(双指针,LintCode,Medium)