困难 · 双指针 · 原题

题目信息

LeetCode #42 · Trapping Rain Water

标签：栈、数组、双指针、动态规划、单调栈

题目要求

给定 n 个非负整数表示每个宽度为 1 的柱子的高度图，计算按此排列的柱子，下雨之后能接多少雨水。

示例 1：

1
2
3
输入：height = [0,1,0,2,1,0,1,3,2,1,2,1]
输出：6
解释：上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图，在这种情况下，可以接 6 个单位的雨水（蓝色部分表示雨水）。

示例 2：

1
2
输入：height = [4,2,0,3,2,5]
输出：9

提示：

  • n == height.length

  • 1 <= n <= 2 * 10^4

  • 0 <= height[i] <= 10^5

代码实现

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
class Solution {
public:
    int trap(vector<int>& height) {
        vector<int> lMaxHeight(height.size(),0),rMaxHeight(lMaxHeight);
        int ans=0,lMax=0,rMax=0;
        for(int i=0;i<height.size();i++){
            lMax=max(height[i],lMax); lMaxHeight[i]=lMax;
        }
        for(int i=height.size()-1;i>=0;i--){
            rMax=max(height[i],rMax); rMaxHeight[i]=rMax;
        }
        for(int i=0;i<height.size();i++){
            ans+=min(lMaxHeight[i],rMaxHeight[i])-height[i];
        }
        return ans;
    }
};

返回目录