Two Pointers - Stack - Trapping Rain Water

https://leetcode.com/problems/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 is able to trap after raining.

原网页有图形来说明。

两个指针。哪个小移动哪一个,同时记录当前位置的雨量。maxLeft cur (left right) maxRight,想象一下,cur 只有同时低于左右两侧的最大值时,才可以储备下雨量。而该雨量又由两者的最小值来决定。lower left right lower

最后会于最高点停止,因此循环在左右相遇时结束即可。第一种方法隐含表明了执行哪个分支,则这个分支的 max 要小于另一个分支。这样看,第二种方法思路更清晰了。

还可以使用 stack. 结合图形,很容易理解文中的方法。两个指针是从高度的角度来考虑的,而 stack 是从宽度的角度来考虑的。从宽度角度来考虑,为了算出面积,需要求出当前的宽度和可以容纳雨水的高度,两者相乘即可。

#include <vector>
#include <algorithm>
using std::vector;
using std::max;

class Solution {
public:
    int trap(vector<int>& height) { // Runtime: 8 ms
        int left = 0, right = height.size() - 1;
        int maxLeft = 0, maxRight = 0;
        int res = 0;
        // while (left <= right) {
        while (left < right) {
            if (height[left] <= height[right]) {
                if (height[left] >= maxLeft) {
                    maxLeft = height[left];
                }
                else {
                    res += maxLeft - height[left];
                }
                ++left;
            }
            else {
                if (height[right] >= maxRight) {
                    maxRight = height[right];
                }
                else {
                    res += maxRight - height[right];
                }
                --right;
            }
        }
        return res;
    }

    int trap2(vector<int>& height) { // Runtime: 8 ms
        int left = 0, right = height.size() - 1;
        int level = 0, water = 0, lower = 0;
        while (left < right) {
            if (height[left] <= height[right]) {
                lower = height[left];
                ++left;
            }
            else {
                lower = height[right];
                --right;
            }
            level = max(level, lower);
            water += level - lower;
        }
        return water;
    }
};
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值