下一个更大元素 II
力扣连接:503. 下一个更大元素 II(中等)
单调栈的方法
加多一个for循环即可,和每日温度题几乎一样
代码
class Solution {
public int[] nextGreaterElements(int[] nums) {
Deque<Integer> stack = new LinkedList<>();
int len = nums.length;
int[] result = new int[len];
Arrays.fill(result, -1);
stack.addLast(0);
for(int i=1;i<len;i++){
while(!stack.isEmpty() && nums[i]>nums[stack.getLast()]){
result[stack.getLast()] = nums[i];
stack.removeLast();
}
stack.add(i);
}
for(int i=0;i<len;i++){
while(!stack.isEmpty() && nums[i]>nums[stack.getLast()]){
result[stack.getLast()] = nums[i];
stack.removeLast();
}
}
return result;
}
}
接雨水
力扣连接:42. 接雨水(中等)
单调栈方法
以下逻辑主要就是三种情况
情况一:当前遍历的元素(柱子)高度小于栈顶元素的高度 height[i] < height[st.top()]
情况二:当前遍历的元素(柱子)高度等于栈顶元素的高度 height[i] == height[st.top()]
情况三:当前遍历的元素(柱子)高度大于栈顶元素的高度 height[i] > height[st.top()]
图解步骤
关键点:
- 遇到相同高度的柱子。更新栈内下标,就是将栈里元素(旧下标)弹出,将新元素(新下标)加入栈中。
代码
class Solution {
public int trap(int[] height) {
Deque<Integer> stack = new LinkedList<>();
int len = height.length;
int sum = 0;
stack.addLast(0);
for(int i=1;i<len;i++){
if(height[i]<height[stack.getLast()]){
stack.addLast(i);
}else if(height[i]==height[stack.getLast()]){
stack.removeLast();
stack.addLast(i);
}else{
while(!stack.isEmpty() && height[i]>height[stack.getLast()]){
int mid = stack.getLast();
stack.removeLast();
if(!stack.isEmpty()){
int h = Math.min(height[i], height[stack.getLast()])-height[mid];
int w = i-stack.getLast()-1;
sum += h*w;
}
}
stack.addLast(i);
}
}
return sum;
}
}
动态规划方法
分别求当前水柱 左边和右边的最大值。
代码
class Solution {
public int trap(int[] height) {
int len = height.length;
int[] left = new int[len];
int[] right = new int[len];
left[0] = 0;
for(int i=1;i<len;i++){
left[i] = Math.max(height[i-1], left[i-1]);
}
right[len-1] = 0;
for(int i=len-2;i>=0;i--){
right[i] = Math.max(height[i+1], right[i+1]);
}
int result = 0;
for(int i=0;i<len;i++){
int min = Math.min(left[i], right[i]);
if(height[i]<min){
result += min-height[i];
}
}
return result;
}
}