LeetCode42. [H]接雨水(动态规划、单调栈、双指针*)
LeetCode42.接雨水思路
对于每一个柱子,他所能接的雨水就是左边最高的柱子和右边最高的柱子小的那一个再减去自己的高度。
既:min(leftMax,rightMax)−height
解题方法
动态规划
官方题解(截取):
创建两个长度为 n 的数组 leftMax 和 rightMax。对于 0≤iheight,则得到一个可以接雨水的区域,该区域的宽度是 i−left−1,高度是 min(height,height)−height,根据宽度和高度即可计算得到该区域能接的雨水量。</ul>代码:
class Solution {
public:
int trap(vector<int>& height) {
int n = height.size();
if (n == 0) {
return 0;
}
vector<int> leftMax(n);
//获取每个元素左边的最大值(从左到右遍历)
leftMax = height;
for (int i = 1; i < n; ++i) {
leftMax = max(leftMax, height);
}
//获取每个元素右边的最大值(从右到左遍历)
vector<int> rightMax(n);
rightMax = height;
for (int i = n - 2; i >= 0; --i) {
rightMax = max(rightMax, height);
}
int ans = 0;
for (int i = 0; i < n; ++i) {
ans += min(leftMax, rightMax) - height;
}
return ans;
}
};class Solution {
public:
int trap(vector<int>& height) {
int ans = 0;
stack<int> stk;
int n = height.size();
//遍历每一个元素
for (int i = 0; i < n; ++i) {
//如果栈中有元素,且当前的柱子高于栈顶,既满足 left<=top<i
while (!stk.empty() && height > height) {
int top = stk.top();
stk.pop();
//防止栈中只有一个元素
if (stk.empty()) {
break;
}
int left = stk.top();
int currWidth = i - left - 1;
int currHeight = min(height, height) - height;
ans += currWidth * currHeight;
}
//i<=top 入栈
stk.push(i);
}
return ans;
}
};复杂度分析
[*]时间复杂度:O(n),其中 n 是数组 height 的长度。两个指针的移动总次数不超过 n。
[*]空间复杂度:O(1)。只需要使用常数的额外空间。
对于官方题解的理解
动态规划的做法中,需要维护两个数组 leftMax 和 rightMax,因此空间复杂度是 O(n)。是否可以将空间复杂度降到 O(1)?注意到下标 i 处能接的雨水量由 leftMax 和 rightMax 中的最小值决定。由于数组 leftMax 是从左往右计算,数组 rightMax 是从右往左计算,因此可以使用双指针和两个变量代替两个数组。维护两个指针 left 和 right,以及两个变量 leftMax 和 rightMax,初始时 left=0,right=n−1,leftMax=0,rightMax=0。指针 left 只会向右移动,指针 right 只会向左移动,在移动指针的过程中维护两个变量 leftMax 和 rightMax 的值。当两个指针没有相遇时,进行如下操作:使用 height 和 height 的值更新 leftMax 和 rightMax 的值;如果 height
页:
[1]