Leetcode 42:接雨水(单调栈)

题目:

给定 n 个非负整数表示每个宽度为 1 的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。

上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图,在这种情况下,可以接 6 个单位的雨水(蓝色部分表示雨水)。 感谢 Marcos 贡献此图。

示例:

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

答案:

看了这个答案才会做的,代码如下:

class Solution {
    public int trap(int[] height) {
        if(height==null||height.length<3){
            return 0;
        }
        int result = 0;
        LinkedList stack = new LinkedList<>();//放index
        for(int i=0;i=height[stack.peek()]){
                int index = stack.pop();
                if(stack.isEmpty()){
                    break;
                }
                result = result + (i-stack.peek()-1)*(Math.min(height[i],height[stack.peek()])-height[index]);
                 
            }
            stack.push(i);
        }
        return result;
    }
}

 

你可能感兴趣的:(LeetCode)