您的位置:首页 > 移动开发

Trapping Rain Water Array+技巧

2015-07-10 22:24 423 查看
思路:

数组+技巧。时间复杂度O(N),空间复杂度O(1)。

找到最高的柱子的位置(可能不止一个,选任意一个都行),这样,将array分成左右两部分:

(1)遍历左部分,保存一个当前认为的最高的柱子的高度h,如果所经过的柱子高度
h'
h
小,那么,
h-h'
一定是被trap住的;时刻更新当前认为的最高柱子。

(2)对于右半部分也是一样。

class Solution {
public:
    int trap(vector<int>& height) {
        int n = height.size();
        //find max height to split the array
        int max = 0;
        for(int i = 0; i < n; ++i) {
            if(height[i] > height[max]) {
                max = i;
            }
        }
        //calculate sum from left to max
        int sum = 0, peak = 0;
        for(int i = 0; i < max; ++i) {
            if(height[i] > peak) {
                peak = height[i];
            }else {
                sum += peak - height[i];
            }
        }
        peak = 0;
        //calculate sum from right to max
        for(int i = n - 1; i > max; --i) {
            if(height[i] > peak) {
                peak = height[i];
            }else {
                sum += peak - height[i];
            }
        }
        return sum;
    }
};
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: