单调栈第二天
目录
503.下一个更大元素II
给定一个循环数组（最后一个元素的下一个元素是数组的第一个元素），输出每个元素的下一个更大元素。数字 x 的下一个更大的元素是按数组遍历顺序，这个数字之后的第一个比它更大的数，这意味着你应该循环地搜索它的下一个更大的数。如果不存在，则输出 -1。
示例 1:
- 输入: [1,2,1]
- 输出: [2,-1,2]
- 解释: 第一个 1 的下一个更大的数是 2；数字 2 找不到下一个更大的数；第二个 1 的下一个最大的数需要循环搜索，结果也是 2。
看到题目的第一想法
利用昨天的单调栈的思路来写
但是它是循环数组该怎么解决？
遍历完一次后再遍历一次，将栈中剩余的弹出
最后剩余的赋值为-1
成功做出来
看到代码随想录的想法
没有我这么麻烦，for循环范围放大为2*length，nums[i%length]取模，逻辑和之前相同
若当前比栈里的小则直接存入
若当前比栈里的大，则弹出，栈顶下标对应的第一个最大值为栈里的值
自己实现时出现的困难
把result全部赋值为-1 自己就不用处理-1了
class Solution {
public int[] nextGreaterElements(int[] nums) {
//也是用单调栈来写？
//for循环两次到末尾
Stack<Integer> stack = new Stack<>();
//需要遍历到当前下标的前一个位置？
stack.push(0);
int[] result = new int[nums.length];
for(int i=1;i<nums.length;i++){
if(nums[stack.peek()]>=nums[i]){
stack.push(i);
}else{
while(!stack.isEmpty()&&nums[stack.peek()]<nums[i]){
result[stack.peek()] = nums[i];
stack.pop();
}
//不要忘了把这个值也push进去
stack.push(i);
}
}
//此时stack中残留的元素:
//1 nums[i]末尾的元素
//2 比较大的几个元素
if(!stack.isEmpty()){
for(int i=0;i<nums.length;i++){
while(!stack.isEmpty()&&nums[stack.peek()]<nums[i]){
//弹出来的下标对应result 的下标
result[stack.peek()] = nums[i];
stack.pop();
}
}
}
while(!stack.isEmpty()){
result[stack.peek()] = -1;
stack.pop();
}
return result;
}
}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 个单位的雨水（蓝色部分表示雨水）。
看到题目的第一想法
用单调栈，没想清楚处理逻辑，要怎么计算雨水
看到代码随想录的第一想法
用单调栈，
遇到第一个比栈顶大的元素时，弹出栈顶做记录，再利用当前遍历到的值计算差值和下标差值
栈顶值(中间值)和栈顶的第二个值(前一个比栈顶大的)，当前下标(当前比栈顶大的)
每次计算的是以栈顶值为底部能存放的雨水容量，累加起来
选择两边的最小值，然后再与栈底的差值做计算
自己实现时遇到的困难
if(height[i]==height[stack.peek()])
遍历时，当遇到栈顶与当前数值相等时，进栈的应该是第二个元素，因为要考虑下标
class Solution {
public int trap(int[] height) {
//卡哥思路 ，单调栈
//遇到第一个比当前大的，则弹出第一个，记录，再弹出第二个，取弹出的第二个和当前的最小值，同时下标的差值进行雨水面积计算
Stack<Integer> stack = new Stack<>();
int startIndex = 0;
while(startIndex<height.length-1&&height[startIndex]<=height[startIndex+1]){
startIndex++;
}
if(startIndex==height.length-1){
return 0;
}
stack.push(startIndex);
int maxArea = 0;
for(int i=startIndex+1;i<height.length;i++){
if(height[i]<height[stack.peek()]){
stack.push(i);
}else if(height[i]==height[stack.peek()]){
//不能直接continue，或者说如果前面和后面相等，必须push后面的那个，比如说
// [9,6,8,8,5,6,3] 这两个相等的8 必须push 后面这个8的下标
//弹出前面那个
stack.pop();
stack.push(i);
continue;
}else{
//弹出第一个栈顶,记录
//用第二个栈顶来进行计算，取当前heit[i]和第二个栈顶的最小值，来进行计算
while(!stack.isEmpty()&&height[stack.peek()]<height[i]){
//弹出第一个
int topOne = stack.pop();
//高度为 当前高度(第二低点，于当前选中元素的最小值)减去 最低点
if(!stack.isEmpty()){
int h = Math.min(height[i],height[stack.peek()])-height[topOne];
maxArea += h*(i-stack.peek()-1);
}
}
stack.push(i);
}
}
return maxArea;
}
}