503.下一个更大元素II
力扣题目链接
给定一个循环数组(最后一个元素的下一个元素是数组的第一个元素),输出每个元素的下一个更大元素。数字 x 的下一个更大的元素是按数组遍历顺序,这个数字之后的第一个比它更大的数,这意味着你应该循环地搜索它的下一个更大的数。如果不存在,则输出 -1。
示例 1:
- 输入: [1,2,1]
- 输出: [2,-1,2]
- 解释: 第一个 1 的下一个更大的数是 2;数字 2 找不到下一个更大的数;第二个 1 的下一个最大的数需要循环搜索,结果也是 2。
提示:
- 1 <= nums.length <= 10^4
- -10^9 <= nums[i] <= 10^9
和每日温度差不多
主要考虑怎么循环,遍历的时候走两遍nums.length*2
再取i会越界, 所以用i%nums.length
模拟转一圈的过程
class Solution {
public int[] nextGreaterElements(int[] nums) {
int[] res=new int[nums.length];
Arrays.fill(res,-1);
Deque<Integer> stack=new LinkedList<>();
if(nums.length==0) return res;
stack.push(0);
for(int i=0;i<nums.length*2;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. 接雨水
力扣题目链接
给定 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
两个遍历,一个遍历右边最高,一个遍历左边最高
取连个中间小的, 再减去本列高度 就是能接的高度 再*1 就是体积
class Solution {
public int trap(int[] height) {
int sum=0;
for(int i=0;i<height.length;i++){
if(i==0||i==height.length-1) continue;
int rheight=height[i];
int lheight=height[i];
for(int r=i+1;r<height.length;r++){
if(height[r]>rheight) rheight=height[r];
}
for(int l=i-1;l>=0;l--){
if(height[l]>lheight) lheight=height[l];
}
int h=Math.min(lheight,rheight)-height[i];
if(h>0) sum+=h;
}
return sum;
}
}
用单调递增栈
思路其实和上面是一样的 也是求左边大的和右边大的
class Solution {
public int trap(int[] height) {
Deque<Integer> stack=new LinkedList<>();
stack.push(0);
int sum=0;
for(int i=0;i<height.length;i++){
while(!stack.isEmpty()&&height[i]>height[stack.peek()]){
int mid=stack.peek(); //存一下中间元素
stack.pop(); //弹出才能取第二个元素
if(!stack.isEmpty()){
int left=stack.peek();
int h=Math.min(height[left],height[i])-height[mid];
int w=i-left-1;
int hold=h*w;
if(hold>0) sum+=hold;
}
}
stack.push(i);
}
return sum;
}
}