LeetCode热题100——接雨水

方法一：动态规划

每个位置能接多少水，取决于，左右边界中最小值和当前高度的差值，所以我们需要找到i的左右最大值

我们可以用两个数组记录每个索引的左右最大值，使用一维动态规划，分别从左、右开始遍历，记录最大值

 

class Solution {
    public int trap(int[] height) {
        int n = height.length;
        if(n == 0){
            return 0;
        }
        int[] leftMax = new int[n]; 
        leftMax[0] = height[0];
        for(int i = 1;i < n;i++){
            leftMax[i] = Math.max(height[i],leftMax[i - 1]);
        }
        int[] rightMax = new int[n];
        rightMax[n - 1] = height[n - 1];
        for(int i = n - 2;i >=0; i--) {
            rightMax[i] = Math.max(height[i],rightMax[i + 1]);
        }
        int ans = 0;
        for(int i = 0;i < n;i++){
            ans += Math.min(leftMax[i], rightMax[i]) - height[i];
        }
        return ans;
    }
}

方法二：双指针

上一个方法我们使用两个数组维护最大值，我们可以简化为使用两个指针维护左右最大值，并且始终是两者中较小值往前走，因为接水量由较短者决定。

class Solution {
    public int trap(int[] height) {
        int ans = 0;
        int left = 0,right = height.length - 1;
        int leftMax = 0,rightMax = 0;
        while(left < right){
            leftMax = Math.max(leftMax,height[left]);
            rightMax = Math.max(rightMax,height[right]);
            if(leftMax < rightMax){
                ans += leftMax - height[left];
                left++;
            } else{
                ans += rightMax - height[right];
                right--;
            }
        }
        return ans;
    }
}