503.下一个更大元素II
1.题目:
给定一个循环数组 nums ( nums[nums.length - 1] 的下一个元素是 nums[0] ),返回 nums 中每个元素的 下一个更大元素 。
数字 x 的 下一个更大的元素 是按数组遍历顺序,这个数字之后的第一个比它更大的数,这意味着你应该循环地搜索它的下一个更大的数。如果不存在,则输出 -1 。
示例 1:
输入: nums = [1,2,1]
输出: [2,-1,2]
解释: 第一个 1 的下一个更大的数是 2;
数字 2 找不到下一个更大的数;
第二个 1 的下一个最大的数需要循环搜索,结果也是 2。
示例 2:
输入: nums = [1,2,3,4,3]
输出: [2,3,4,-1,4]
2.思路:
循环数组
1.将两个nums数组拼接在一起,使用单调栈计算出每一个元素的下一个最大值,最后再把结果集即result数组resize到原数组大小就可以了。扩充nums数组相当于多了一个O(n)的操作,复杂度高
2.(推荐)遍历的过程中模拟走两边nums。
3.代码:
public int[] nextGreaterElements(int[] nums) {
if(nums.length<=1|| nums==null){
return new int[]{-1};
}
int[] res=new int[nums.length];
Stack<Integer> stack=new Stack<>();
Arrays.fill(res,-1);
for(int i=0;i<2*nums.length;i++){
while(!stack.isEmpty() && nums[i%nums.length]>nums[stack.peek()]){
res[stack.peek()]=nums[i%nums.length];
stack.pop();
}
stack.push(i%nums.length);
}
return res;
}
42. 接雨水
1.题目:
给定 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
https://leetcode.cn/problems/trapping-rain-water
2.思路:
双指针法
只要记录左边柱子的最高高度 和 右边柱子的最高高度,就可以计算当前位置的雨水面积,这就是通过列来计算。
当前列雨水面积:min(左边柱子的最高高度,记录右边柱子的最高高度) - 当前柱子高度。
为了得到两边的最高高度,使用了双指针来遍历,每到一个柱子都向两边遍历一遍,这其实是有重复计算的。我们把每一个位置的左边最高高度记录在一个数组上(maxLeft),右边最高高度记录在一个数组上(maxRight),这样就避免了重复计算。
当前位置,左边的最高高度是前一个位置的左边最高高度和本高度的最大值。
即从左向右遍历:maxLeft[i] = max(height[i], maxLeft[i - 1]);
从右向左遍历:maxRight[i] = max(height[i], maxRight[i + 1]);
单调栈法
1.首先单调栈是按照行方向来计算雨水,如图:
2.使用单调栈内元素的顺序
从栈头(元素从栈头弹出)到栈底的顺序应该是从小到大的顺序。
一旦发现添加的柱子高度大于栈头元素了,此时就出现凹槽了,栈头元素就是凹槽底部的柱子,栈头第二个元素就是凹槽左边的柱子,而添加的元素就是凹槽右边的柱子。
如图:
- 遇到相同高度的柱子怎么办。
遇到相同的元素,更新栈内下标,就是将栈里元素(旧下标)弹出,将新元素(新下标)加入栈中。
例如 5 5 1 3 这种情况。如果添加第二个5的时候就应该将第一个5的下标弹出,把第二个5添加到栈中。
因为我们要求宽度的时候 如果遇到相同高度的柱子,需要使用最右边的柱子来计算宽度。
如图所示:
- 栈里要保存什么数值
使用单调栈,也是通过 长 * 宽 来计算雨水面积的。
长就是通过柱子的高度来计算,宽是通过柱子之间的下标来计算,
那么栈里有没有必要存一个pair<int, int>类型的元素,保存柱子的高度和下标呢。
其实不用,栈里就存放下标就行,想要知道对应的高度,通过height[stack.top()] 就知道弹出的下标对应的高度了。
单调栈处理逻辑
以下操作过程其实和 739. 每日温度 (opens new window)也是一样的,建议先做 739. 每日温度 (opens new window)。
以下逻辑主要就是三种情况
- 情况一:当前遍历的元素(柱子)高度小于栈顶元素的高度 height[i] < height[st.top()]
- 情况二:当前遍历的元素(柱子)高度等于栈顶元素的高度 height[i] == height[st.top()]
- 情况三:当前遍历的元素(柱子)高度大于栈顶元素的高度 height[i] > height[st.top()]
先将下标0的柱子加入到栈中,st.push(0);
。 栈中存放我们遍历过的元素,所以先将下标0加进来。
然后开始从下标1开始遍历所有的柱子,for (int i = 1; i < height.size(); i++)
。
如果当前遍历的元素(柱子)高度小于栈顶元素的高度,就把这个元素加入栈中,因为栈里本来就要保持从小到大的顺序(从栈头到栈底)。
如果当前遍历的元素(柱子)高度等于栈顶元素的高度,要跟更新栈顶元素,因为遇到相相同高度的柱子,需要使用最右边的柱子来计算宽度。
如果当前遍历的元素(柱子)高度大于栈顶元素的高度,此时就出现凹槽了,如图所示:
取栈顶元素,将栈顶元素弹出,这个就是凹槽的底部,也就是中间位置,下标记为mid,对应的高度为height[mid](就是图中的高度1)。
此时的栈顶元素st.top(),就是凹槽的左边位置,下标为st.top(),对应的高度为height[st.top()](就是图中的高度2)。
当前遍历的元素i,就是凹槽右边的位置,下标为i,对应的高度为height[i](就是图中的高度3)。
此时大家应该可以发现其实就是栈顶和栈顶的下一个元素以及要入栈的元素,三个元素来接水!
那么雨水高度是 min(凹槽左边高度, 凹槽右边高度) - 凹槽底部高度,代码为:int h = min(height[st.top()], height[i]) - height[mid];
雨水的宽度是 凹槽右边的下标 - 凹槽左边的下标 - 1(因为只求中间宽度),代码为:int w = i - st.top() - 1 ;
当前凹槽雨水的体积就是:h * w
。
3.代码:
public int trap(int[] height) {
//双指针
if(height.length<=2){
return 0;
}
int[] maxleft=new int[height.length];
int[] maxright=new int[height.length];
//更新每个柱子左边的最大高度
maxleft[0]=height[0];
for(int i=1;i<height.length;i++){
maxleft[i]=Math.max(height[i],maxleft[i-1]);
}
//更新每个柱子右边的最大高度
maxright[height.length-1]=height[height.length-1];
for(int j=height.length-2;j>=0;j--){
maxright[j]=Math.max(height[j],maxright[j+1]);
}
int sum=0;
for(int i=0;i<height.length;i++){
int h=Math.min(maxleft[i],maxright[i])-height[i];
sum+=h;
}
return sum;
}
public int trap(int[] height) {
//单调栈,栈内存放遍历元素的下标
if(height.length<=2){
return 0;
}
Stack<Integer> stack=new Stack<>();
stack.push(0);
int sum=0;
for(int i=1;i<height.length;i++){
if(height[i]<height[stack.peek()]){
stack.push(i);
}else if(height[i]==height[stack.peek()]){
stack.pop();
stack.push(i);
}else{
while(!stack.isEmpty() &&height[i]>height[stack.peek()]){
int mid=stack.peek();
stack.pop();
if(!stack.isEmpty()){
int h=Math.min(height[stack.peek()],height[i])-height[mid];
int w=i-stack.peek()-1;
int res=h*w;
sum+=res;
}
}
stack.push(i);
}
}
return sum;
}