接雨水问题
给定 n n n个非负整数表示每个宽度为1的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。
解题思路:每一个位置的雨水接收量大小取决于该位置两端的最大柱子高度,当知道一个位置
i
i
i的柱子高度
H
i
H_{i}
Hi和这个位置左右两边的最大柱子高度
H
L
i
HL_{i}
HLi,
H
R
i
HR_{i}
HRi后,该位置的雨水接收量满足:
V
i
=
m
i
n
(
H
L
i
,
H
R
i
)
−
H
i
V_{i} = min(HL_{i}, HR_{i}) - H_{i}
Vi=min(HLi,HRi)−Hi
则对应的总的雨水接收量
V
N
V_{N}
VN可直接累加计算:
V
N
=
∑
i
=
1
n
V
i
V_{N} = \sum^{n}_{i=1}V_{i}
VN=i=1∑nVi
于是解决问题的关键变成了求解每个位置两端的最大柱子高度,而这个问题可以使用动态规划分别求解出数组
H
L
HL
HL和
H
R
HR
HR。对于数组
H
L
HL
HL,第
i
i
i个位置的元素值表示该位置左侧存在的最大柱子高度(可以是其本身),对于数组HR,第
i
i
i个位置的元素值表示该位置右侧存在的最大柱子高度(可以是其本身),具体C++代码实现如下:
int trap(vector<int>& height) {
vector<int> dpl(height.size()+1, 0); // 表示HL
vector<int> dpr(height.size()+1, 0); // 表示HR
dpl[0] = 0, dpr[height.size()] = 0;
for(int i=0; i<height.size(); i++){
dpl[i+1] = (dpl[i] > height[i] ? dpl[i] : height[i]);
}
for(int i=height.size()-1; i>=0; i--){
dpr[i] = (dpr[i+1] > height[i] ? dpr[i+1] : height[i]);
}
int tsum = 0;
for(int i=0; i<height.size(); i++){
tsum += ((dpl[i+1] > dpr[i] ? dpr[i] : dpl[i+1])-height[i]);
}
return tsum;
}