给定 n 个非负整数表示每个宽度为 1 的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。
上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图,在这种情况下,可以接 6 个单位的雨水(蓝色部分表示雨水)。 感谢 Marcos 贡献此图。
示例:
输入: [0,1,0,2,1,0,1,3,2,1,2,1]
输出: 6
看了看leecode上的最快速度的代码真的是丑的没办法…
还不如我这个,速度同样最快,思路清晰,代码还比较优雅hhh
主要也就是维护两个表示左右两边最大高度的数组,如果存在左边也比当前高,右边也比当前高,那么ans加上与左右最小高度的差值就好了。
static const auto __=[](){
ios::sync_with_stdio(0);
cin.tie(nullptr);
return nullptr;
}();
class Solution {
public:
int trap(vector<int>& height) {
vector<int> left;
vector<int> right;
for (int i = 0; i < height.size(); ++i)
{
left.push_back(height[i]);
right.push_back(height[i]);
}
for (int i = 1; i < left.size(); ++i)
{
if (left[i-1] > left[i])
{
left[i] = left[i-1];
}
}
for (int i = height.size() - 2; i >= 0; --i)
{
if (right[i+1] > right[i])
{
right[i] = right[i+1];
}
}
int ans = 0;
for (int i = 0; i < height.size(); ++i)
{
if (left[i] >= height[i] && right[i] >= height[i])
{
ans += min(left[i], right[i]) - height[i];
}
}
return ans;
}
};