Trapping Rain Water(leetcode hard)

这是自己在面试微软过程中遇到的编程题,印象比较深刻是因为当时自己尝试想了很多方法都是不对的。

思想:考虑在每一个位置,它可以存多少水。而这取决于它左边最高的板子,它右边最高的板子,这两者的最小值高于当前位置多少。

下面的代码里,forward和backward分别记录的是包括当前位置的板子的最高值。仔细想了下,是否包括当前位置,都是OK的。

class Solution {
public:
    int trap(vector<int>& height) {
        int n = height.size();
        if(n==0) return 0;
        int forward[n];
        int backward[n];
        forward[0] = height[0];
        for(int i=1;i<n;i++){
            forward[i] = max(height[i], forward[i-1]);
        }
        backward[n-1] = height[n-1];
        for(int i=n-2;i>=0;i--){
            backward[i] = max(height[i], backward[i+1]);
        }
        int ans = 0;
        for(int i=0;i<n;i++){
            int tmp = min(forward[i],backward[i]);
            if(tmp>height[i])
                ans += tmp-height[i];
        }
        return ans;
    }
};
// https://www.cnblogs.com/AndyJee/p/4821590.html
class Solution {
public:
    int trap(vector<int>& height) {
        int n = height.size();
        int ans = 0;
        if(n==0) return ans;
        int leftmost = height[0], rightmost = height[n-1];
        int l = 0, r = n-1;
        while(l<=r){
            if(leftmost <= rightmost){  // leftmost是短板
               ans += leftmost-height[l];
               l++;
               if(l<=r)
               leftmost = max(leftmost, height[l]);
            }else{        // rightmost是短板
               ans += rightmost - height[r];
               r--;
               if(l<=r)
               rightmost = max(rightmost, height[r]);
            }
        }
        return ans;
    }
};

单调栈解法:http://www.cnblogs.com/grandyang/p/8887985.html

class Solution {
public:
    int trap(vector<int>& height) {
        int n = height.size();
        stack<int> mystack;  // 单调递减栈 (非严格)
        int ans = 0;
        for(int i=0;i<n;i++){
            if(mystack.empty()){
                mystack.push(i);
            }else{
                if( height[mystack.top()]>=height[i]){
                    mystack.push(i);
                }else{
                    if( mystack.size()>=2){  // 形成一个坑
                        int middle = mystack.top();
                        mystack.pop();
                        int left = mystack.top();
                        int right = i;
                        ans += (min(height[left],height[right])-height[middle])*(right-left-1);
                        i--;
                    }else{
                        mystack.pop();
                        i--;
                        
                    }
                }
            }
        }
        return ans;
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值