leetcode#42 Trapping rain water的五种解法详解

leetcode#42 Trapping rain water

这道题十分有意思，可以用很多方法做出来，每种方法的思想都值得让人细细体会。

42. 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.

For example,
Given [0,1,0,2,1,0,1,3,2,1,2,1], return 6.

 Solution 1:

通过分别计算每一坐标i上有多少水，进而将其相加得到答案。

问题是我们如何知道每一坐标i上有多少水呢？仔细思考，其实只有出现“两高夹一矮”才可能会存到水，如下图所示。

进而，我们可以想到：每一坐标i上存多少水是由 1.其自身高度 2.它左边的最高高度left_most 3.它右边的最高高度right_most三种因素决定的。

当 min{ left_most, right_most} 小于或等于其自身高度时，它能存的水就是0，比如array[1]=1,其left_most= array[0]=0, 其right_most=array[7]=3, min{left_most, right_most}=left_most=0< height= array[1]=1，这也就是说坐标1 存不了水。

当min{ left_most,right_most} 大于其自身高度时，这时这三者间出现了“两高夹一矮”的情况，故其能存水，而且其存水数= min{left_most,right_most} - height。

我们分别来对一些坐标进行验证：

坐标1，存水数=0.//正确

坐标2，leff_most=1,right_most=3，存水数=left_most-height=1-0=1.//正确

坐标3，left_most=1,right_most=3,min{left_most,right_most}=1=height，存水数=0.//正确

读者可以对每个坐标进行验证，会发现以上结论皆是正确的。所以，现在我们的solution就出来了，我们只需要求出每个坐标对应的left_most和right_most，再把存水数相加，就是总的存水数了。

所以，很朴素自然的一个想法就是，遍历一遍数组，对每个数组元素遍历左边一次求出left_most，遍历右边一次求出right_most。

代码如下，

 

//29ms 6.36%
//complexity: O(N^2)
int trap(vector<int>& height)
{
int ans = 0;
int size = height.size();
for (int i = 1; i < size - 1; i++) {
int max_left = 0, max_right = 0;
for (int j = i; j >= 0; j--) { //Search the left part for max bar size
max_left = max(max_left, height[j]);
}
for (int j = i; j < size; j++) { //Search the right part for max bar size
max_right = max(max_right, height[j]);
}
ans += min(max_left, max_right) - height[i];
}
return ans;
}

 

Solution 2：

在solution 1里，我们已经知道只要求出left_most和right_most，就可以求出答案，那能不能优化一下求这两个数的过程呢？当然是可以的，我们只需要左遍历一次数组，右遍历一次数组，即可得到left_most和right_most。

/*Solution2: 上一种方法其实有优化的空间
通过两次for循环可分别求得left_most和right_most，第三次for循环即可求得sum，
complexity: O(n)
*/
int trap(vector<int>& height)
{
    if(height == null)
        return 0;
    int ans = 0;
    int size = height.size();
    vector<int> left_max(size), right_max(size);
    left_max[0] = height[0];
    for (int i = 1; i < size; i++) {
        left_max[i] = max(height[i], left_max[i - 1]);
    }
    right_max[size - 1] = height[size - 1];
    for (int i = size - 2; i >= 0; i--) {
        right_max[i] = max(height[i], right_max[i + 1]);
    }
    for (int i = 1; i < size - 1; i++) {
        ans += min(left_max[i], right_max[i]) - height[i];
    }
    return ans;
}

 

 

Solution 3：

这里再介绍一种优化方法，双指针法，在数组首尾分别创建一个指针，两指针相见时结束循环。

int trap(vector<int>& height)
{
    int left = 0, right = height.size() - 1;
    int ans = 0;
    int left_max = 0, right_max = 0;
    while (left < right) {
        if (height[left] < height[right]) {
            height[left] >= left_max ? (left_max = height[left]) : ans += (left_max - height[left]);
            ++left;
        }
        else {
            height[right] >= right_max ? (right_max = height[right]) : ans += (right_max - height[right]);
            --right;
        }
    }
    return ans;
}

 

Solution 4：

既然可以纵向的求存水数，那我们能不能一层一层的求存水数呢？

这是第一层，当我们遇到一个空的，且不在边界，存水数+1，所以第一层我们在i=2,i=5 时分别+1.

第二层，存水数+4，依次类推，最终可以求出答案。

代码笔者就不给了，读者有兴趣的可以自己写来试试。

 

Soluton 5：

这是在leetcode中solution给出的一种很新颖的解法，利用了栈的结构，通过维护一个非递增栈来得到答案。

本质思想还是利用了要存水必须是“两高夹一矮”这个特点，只不过这里是用非递增栈来实现。

下面定义一些符号以便理解：

stack[-1] 栈顶元素

stack[-2] 栈顶的下面一个元素（即倒数第二个元素）

solution4的整个算法是这么实现的：遍历数组，遇到一个元素时，将其与栈顶元素比较，如果其小于等于栈顶元素，直接压栈，将其放入栈中（为维护非递增栈的结构,不能将比栈顶元素大的元素压栈），

若是其大于栈顶元素，此时一定形成了一个“两高夹一矮”局面，因为栈是非递增栈，所以 stack[-1]<stack[-2],又 current>stack[-1],所以是一个“两高夹一矮”局面，此时算完存水数后栈顶元素出栈，继续判断，

递归处理即可。

在上例中整个过程是这样的。

step0: 0不入栈

step1: 1>0 array[1] 入栈 栈：[1]

step2: 0<stack[-1]=1 入栈 栈：[1,0]

step3: 2>stack[-1]=0 存水数+1，0出栈，2>stack[-1]=1, 此时stack内元素不足2，不足以形成“两高夹一矮”局面, 1出栈，2入栈 栈：[2]

step4: 1<stack[-1]=2 1入栈 栈：[2,1]

step5: 0<stack[-1]=1 0入栈 栈：[2,1,0]

step6: 1>stack[-1]=0 存水数+1，0出栈 1=stack[-1] 1入栈 栈：[2,1,1] 

step7: 3>stack[-1]=1 存水数+0，1出栈 3>stack[-1]=1 存水数+3,1出栈 3>stack[-1]=2 存水数+0 2出栈 3入栈 栈：[3]

step8: 2<stack[-1] 2入栈 栈：[3,2]

step9: 1<stack[-1] 1入栈 栈：[3,2,1]

step10: 2>stack[-1] 存水数+1 1出栈 2入栈 栈：[3,2,2]

step 11:1<stack[-1] 入栈 栈：[3,2,2,1] 

done

/*Solution4
Stack solution
这个solution利用了栈结构，通过维护一个非递增栈，一步一步算出ans
*/

int trap(vector<int>& height)
{
    int ans = 0, current = 0;
    stack<int> st;
    while (current < height.size()) {
        while (!st.empty() && height[current] > height[st.top()]) {
            int top = st.top();
            st.pop();
            if (st.empty())
                break;
            int distance = current - st.top() - 1;
            int bounded_height = min(height[current], height[st.top()]) - height[top];
            ans += distance * bounded_height;
        }
        st.push(current++);
    }
    return ans;
}