【难度:难】
给定 n
个非负整数表示每个宽度为 1
的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。
示例 1:
输入: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:
输入:height = [4,2,0,3,2,5] 输出:9
提示:
n == height.length
1 <= n <= 2 * 10^4
0 <= height[i] <= 10^5
思路:根据图可以找到规律:当前指针指向下标t时,求t所能接到雨水的高度,由min(t左边最高,t右边最高)-height[t]所决定,eg:上面例子中下标t=2(下标从零开始),时左边最高为1,右边最高为3,所以所能接的雨水为min(1,3)-height[2]=1-0=1;
那么如何快速找到当前下标左右两边的最大高度呢?暴力求解显然时间复杂度过高,这里可以采用动态规划:将一个问题划分为若干个子问题,最终子问题的求和就是该问题的解。
本题可以分开求左右两边的最大值:
(1)遍历一遍数组,记录每个下标左边的最大值LeftMaxheight[]:初始状态为LeftMaxheight[0]=height[0];
对于后面的每一个下标t有LeftMaxheight[t]=max(LeftMaxheight[t-1],height[t])
(2)同理逆向遍历数组,记录每个下标右边边的最大值RightMaxheight[]:初始状态为RightMaxheight[n-1]=height[n-1];
对于后面的每一个下标t有RightMaxheight[t]=max(RightMaxheight[t+1],height[t]) (注意:因为是逆向,所以先知道t+1的右边最大高度,才能知道t的右边最大高度)
参考:
class Solution {
public:
int trap(vector<int>& height) {
const int n=height.size();
int LeftMaxheight[30000];
int RightMaxheight[30000];
LeftMaxheight[0]=height[0];
RightMaxheight[n-1]=height[n-1];
for(int i=1;i<n;i++){
LeftMaxheight[i]=max(LeftMaxheight[i-1],height[i]);
}
for(int i=n-2;i>=0;i--){//注意是n-2不然数组越界
RightMaxheight[i]=max(RightMaxheight[i+1],height[i]);
}
int sum=0;
for(int i=1;i<n-1;i++){
int h= min(RightMaxheight[i],LeftMaxheight[i]);
int he= h-height[i];
sum+=he;
}
return sum;
}
};