給定 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;
}
};