Trapping Rain Water

为什么我的思路总是和别人的不一样呢……

具体想法是,设想这些墙都是排成一队的人,那么总有一个最高的墙,在这个墙左边,每个墙能存的雨水就和他向右看能看到的墙的个数有关,直到某一个比他还高的墙将他视线挡住为止。那么,在这两个比较高的墙之间,就是可以存雨水的部分。最高墙的右侧类似,只不过是需要向左看。

因此,首先找到最高的墙是哪个,然后左右分开处理即可。

class Solution {
public:
    int trap(int A[], int n) {
        int index_max = -1;
        int max = INT_MIN;
        for (int i = 0; i < n; ++i)
            if (A[i] > max) {
                index_max = i;
                max = A[i];
            }
        int res = 0;
        int i = 1;
        int tmp_max = A[0];
        while (i < index_max) {
            if (A[i] < tmp_max) {
                res += tmp_max - A[i];
                ++i;
            }
            else {
                tmp_max = A[i];
                ++i;
            }
        }
        i = n - 2;
        tmp_max = A[n - 1];
        while (i > index_max) {
            if (A[i] < tmp_max) {
                res += tmp_max - A[i];
                --i;
            }
            else {
                tmp_max = A[i];
                --i;
            }
        }
        return res;
    }
};

http://oj.leetcode.com/problems/trapping-rain-water/

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值