LeeCode( 动态编程)42_接雨水
题目:
给定 n 个非负整数表示每个宽度为 1 的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。
示例 1:
输入:height = [0,1,0,2,1,0,1,3,2,1,2,1]
输出:6
解释:上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图,在这种情况下,可以接 6 个单位的雨水(蓝色部分表示雨水)。
示例 2:
输入:height = [4,2,0,3,2,5]
输出:9
提示:
n == height.length
0 <= n <= 3 * 104
0 <= height[i] <= 105
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/trapping-rain-water
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路:
方法一:
暴力
- 要想找到数组每个坐标能接的最大的雨水,只需要找到该点左右两边的最高值然后取这两个最高值的最小值减去当前坐标对应的值即可得到当前左边可以接到的最大雨水。
Java代码:
public class 接雨水 {
public int trap(int[] height) {
int ans = 0;
int size = height.length;
for(int i =1;i<size;i++){
int max_left = 0,max_right = 0;
for(int j=i;j>=0;j--){
max_left = Math.max(max_left, height[j]);
}
for(int j=i;j<size;j++){
max_right = Math.max(max_right, height[j]);
}
ans+=Math.min(max_left, max_right)-height[i];
}
return ans;
}
方法二:
动态编程
- 此外可以在上面的基础上进行优化,用动态编程的思想对代码进行修改
public class 接雨水 {
public int trap(int[] height) {
if(height == null || height.length == 0){
return 0;
}
int ans = 0;
int size = height.length;
int left_max[] = new int[size];
int right_max[] = new int[size];
left_max[0] = height[0];
for(int i=1;i<size;i++){
left_max[i] = Math.max(left_max[i-1], height[i]);
}
right_max[size - 1] = height[size - 1];
for (int i = size - 2; i >= 0; i--) {
right_max[i] = Math.max(height[i], right_max[i + 1]);
}
for (int i = 1; i < size - 1; i++) {
ans += Math.min(left_max[i], right_max[i]) - height[i];
}
return ans;
}
}
方法三:
栈
- 我们通过维护一个栈,进行一次遍历后便能计算出结果。
用current表示当前的坐标,当current小于栈顶元素时将current进行入栈操作(该处形成低洼),当遍历到current大于栈顶元素时(即找到了低洼的两个边界)则做出栈操作并计算两个边界的距离并乘以当前坐标的高度与两个边界高度最小值的差值得到低洼面积,最后加入答案ans。
import java.util.Deque;
import java.util.LinkedList;
public class 接雨水 {
public int trap(int[] height) {
int ans = 0,current = 0;
Deque<Integer> stack = new LinkedList<Integer>();
while(current < height.length){
while(!stack.isEmpty() && height[current] > height[stack.peek()]){
int top = stack.pop();
if(stack.isEmpty())
break;
int distance = current - stack.peek()-1;
int dounded_height = Math.min(height[current], height[stack.peek()]) - height[top];
ans += distance*dounded_height;
}
stack.push(current++);
}
return ans;
}
}
方法四:
双指针
- 我们假设两端各有一个很高的条形块,它们之间会形成很多低洼积水。 具体的积水值等于(较低的一端高度-该处高度)。
- 当左端比右端低时(即height[left]<right[right])则从右端开始遍历。否则,从左端开始遍历。
- 我们只要在遍历的时候维护left_max 和 right_max不断更新就可以得到答案。
import java.util.Deque;
import java.util.LinkedList;
public class 接雨水 {
public int trap(int[] height) {
int left = 0,right = height.length-1;
int ans = 0;
int left_max = 0, right_max = 0;
while(left<right){
if (height[left] < height[right]) {
if(height[left]>left_max){
left_max = height[left];
}else{
ans+=left_max-height[left];
}
left++;
}else{
if(height[right]>right_max){
right_max = height[right];
}else{
ans+=right_max-height[right];
}
right--;
}
}
return ans;
}
}