leetcode 42. 接雨水
给定 n 个非负整数表示每个宽度为 1 的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。
示例 1:
输入:height = [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 个单位的雨水(蓝色部分表示雨水)。
示例 2:
输入:height = [4,2,0,3,2,5]
输出:9
提示:
n == height.length
0 <= n <= 3 * 104
0 <= height[i] <= 105
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/trapping-rain-water
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
class Solution { public int trap(int[] height) { int length = height.length; if (length < 3) { return 0; } int st = 0; int end = 0; int sum = 0; while (end < length) { for (int i = st; i < length; i++) { st = i; if (i >= length - 2) { return sum; } if (height[i + 1] < height[i]) { break; } } end = 0; for (int i = st + 2; i < length; i++) { if (height[i] >= height[i - 1]) { if (height[i] > height[st]) { end = i; break; } if (end == 0) { end = i; } else { if (height[i] >= height[end]) { end = i; } } } } if (end == 0) { return sum; } int limit = Math.min(height[st], height[end]); for (int i = st + 1; i < end; i++) { if (limit - height[i] > 0) { sum += (limit - height[i]); } } st = end; } return sum; } }