LeetCode-Trapping Rain Water解法

标签：leetcode   trapping rain water   

解法：（1）建立一个二维数组，记录各位置左侧最高点及右侧最高点，可通过O(n)计算得到。

（2）根据上述信息及当前高度，计算该点竖直方向能容纳多少水，仍是O（n）完成。

综上，渐近时间复杂度为O（n）；

class Solution {
public:
    int trap(int A[], int n) {
if(n<=0)
return 0;
        int* high=new int[n*2];
        high[0*2+0]=0;
        for(int i=1;i<n;i++){
            if(A[i-1]>high[(i-1)*2+0])
                high[i*2+0]=A[i-1];
            else
                high[i*2+0]=high[(i-1)*2+0];
        }
        high[(n-1)*2+1]=0;
        for(int i=n-2;i>=0;i--){
            if(A[i+1]>high[(i+1)*2+1])
                high[i*2+1]=A[i+1];
            else
                high[i*2+1]=high[(i+1)*2+1];
        }
        int sum=0;
        for(int i=0;i<n;i++){
            int tmp=high[i*2+0]>high[i*2+1]?high[i*2+1]:high[i*2+0];
tmp=tmp-A[i];
if(tmp<0)
tmp=0;
            sum+=tmp;
        }
        delete[] high;
        return sum;
    }
};

LeetCode-Trapping Rain Water解法

标签：leetcode   trapping rain water   

原文地址：http://blog.csdn.net/awakewind/article/details/39854295