leetcode 42. 接雨水 hard
题目描述:
给定 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
解题思路:
方法1:
可以按列来计算, 首先,如果按照列来计算的话,宽度一定是1了,我们再把每一列的雨水的高度求出来就可以了。可以看出每一列雨水的高度,取决于该列左侧最高的柱子和右侧最高的柱子中最矮的那个柱子的高度,然后减去自己的高度(注意首尾元素永远不能接雨水)。而求每个元素左右两边的最大值有两种方法:
1) dp
时间复杂度:O(n) 空间复杂度:O(n)
2) 双指针法 (这个方法好,理解,记这个)
1. 对于左边而言,left_max是可信的,而right_max是不可信的,真实的right_max大于等于right_max,对于右边而言,刚好相反
2. 每一列雨水的高度,取决于左右两侧最高那块挡板较小的一个
方法2: 单调栈,但是这个方法用的单调栈,太巧妙了,觉得意义不是很大
代码:
// 双指针法
class Solution {
public:
int trap(vector<int>& height) {
if (height.empty())
return 0;
int left_max = 0, right_max = 0;
int left = 0, right = height.size()-1;
int ret = 0;
while (left <= right){
if (left_max < right_max){
ret += max(0, left_max-height[left]); //必须 max(0, ) 因为left_max 其实可能小于height【i】
left_max = max(left_max, height[left]);
left++;
} else {
ret += max(0, right_max-height[right]);
right_max = max(right_max, height[right]);
right--;
}
}
return ret;
}
};
//dp
class Solution {
public:
int trap(vector<int>& height) {
//首尾元素不能接雨水
if(height.size()<=2)
return 0;
//每个元素左边(右边)最大的高度(包括它自己)
vector<int> left_max(height.size()),right_max(height.size());
left_max[0]=height[0]; // 第一个元素左边最大的就是它自己
right_max.back()=height.back(); //最后一个元素右边最大的就是它自己
for(int i=1;i<left_max.size();++i){
left_max[i]=max(height[i],left_max[i-1]);
}
for(int i=right_max.size()-2;i>=0;--i){
right_max[i]=max(height[i],right_max[i+1]);
}
int res=0;
// 当前位置能接到的雨水 是 min(left_max[i],right_max[i])-height[i]
// 同样首尾元素不能接雨水
for(int i=1;i<=height.size()-2;++i){
res += min(left_max[i],right_max[i])-height[i];
}
return res;
}
};
// 单调栈
class Solution {
public:
int trap(vector<int>& height) {
int n = height.size(), ans = 0;
stack<int> st;
for (int i = 0; i < n; i++) {
while (!st.empty() && height[st.top()] < height[i]) {
int top = st.top();
st.pop();
if (st.empty()) break;
ans += (i - st.top() - 1) * (min(height[st.top()], height[i]) - height[top]);
}
st.push(i);
}
return ans;
}
};