leetcode 42. 接雨水的三种方法——动态规划；双指针；单调栈

最新推荐文章于 2024-09-01 23:39:35 发布

fatfishccc

最新推荐文章于 2024-09-01 23:39:35 发布

阅读量939

点赞数 29

文章标签： leetcode 动态规划算法 java 数据结构

本文链接：https://blog.csdn.net/weixin_44553952/article/details/137973712

版权

题目

leetcode 42. 接雨水

解法一：动态规划

leftMax[i] 表示下标 i 左侧最高柱子高度，rightMax[i]表示下标 i 右侧最高柱子高度

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

时间复杂度：O(n)
空间复杂度：O(n)

方法二：双指针（对动态规划法的改进）

使用 leftMax 和 rightMax 来代替 leftMax[] 和 rightMax[]，是因为左右指针遍历到的地方，能够保证选择当前所处位置能够容纳的雨水（height[l] < height[r] 或者 height[l] >= height[r] 这两种情况采取不同的操作）

class Solution {
    public int trap(int[] height) {
        int n = height.length, l = 0, r = n-1, res = 0;
        if (n <= 2) return 0;
        int leftMax = height[0], rightMax = height[n - 1];
        while (l < r) {
            leftMax = Math.max(leftMax, height[l]);
            rightMax = Math.max(rightMax, height[r]);
            if (height[l] < height[r]) {
                res += leftMax - height[l];
                l++;
            } else {
                res += rightMax - height[r];
                r--;
            }
        }
        return res;
    }
}

时间复杂度：O(n)
空间复杂度：O(1)

方法三：单调栈

统计雨水量的方法与前述两种方法不一样，每次遍历栈顶元素，就记录栈顶对应位置的、高度为min(height[i], height[left])限制的雨水量。

class Solution {
    public int trap(int[] height) {
        int n = height.length, l = 0, r = n-1, res = 0;
        if (n <= 2) return 0;
        Deque<Integer> q = new LinkedList<>();
        for (int i = 0; i < n; i++) {
            while (!q.isEmpty() && height[i] > height[q.peek()]) {
                // 记录栈顶对应位置的，高度为min(height[i], height[left])限制的雨水量
                int top = q.pop();
                if (q.isEmpty()) break;
                int left = q.peek();
                res += (i-left-1) * (Math.min(height[i], height[left]) - height[top]);
            }
            q.push(i);
        }
        return res;
    }
}