Programming

Leetcode-42. Trapping Rain Water

Leetcode-42. Trapping Rain Water

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

Example 1:

Input: height = [0,1,0,2,1,0,1,3,2,1,2,1] Output: 6 Explanation: The above elevation map (black section) is represented by array [0,1,0,2,1,0,1,3,2,1,2,1]. In this case, 6 units of rain water (blue section) are being trapped. Example 2:

Input: height = [4,2,0,3,2,5] Output: 9

<解題>


class Solution {
    public int trap(int[] height) {
        int left = 0, right = height.length-1;
        int left_max = 0, right_max = 0, ans = 0;
        while(left < right){
            if(height[left] < height[right]){
                if(height[left] > left_max){
                    left_max = height[left];
                } else{ //低窪
                    ans += left_max - height[left];
                }
                left++;
            } else{
                if(height[right] > right_max){
                    right_max = height[right];
                } else{
                    ans += right_max - height[right];
                }
                right--;
            }
        }
        return ans;
    }
}

Time: O(n) Space: O(1)