1 下一个更大的元素II
题目
给定一个循环数组 nums ( nums[nums.length - 1] 的下一个元素是 nums[0] ),返回 nums 中每个元素的 下一个更大元素 。
数字 x 的 下一个更大的元素 是按数组遍历顺序,这个数字之后的第一个比它更大的数,这意味着你应该循环地搜索它的下一个更大的数。如果不存在,则输出 -1 。
代码
class Solution {
public int[] nextGreaterElements(int[] nums) {
Stack<Integer> st=new Stack<>();
int[] res=new int[nums.length];
Arrays.fill(res,-1);
st.push(0);
for(int i=1;i<nums.length*2;i++){
while(!st.isEmpty() && nums[i%nums.length]>nums[st.peek()]){
res[st.peek()]=nums[i%nums.length];
st.pop();
}
st.push(i%nums.length);
}
return res;
}
}
总结
只能说很巧妙,但我想不到。
遍历两边nums数组来达到循环
2 接雨水
题目
给定 n 个非负整数表示每个宽度为 1 的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。
代码
//动态规划
class Solution {
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(maxleft[i-1],height[i]);
maxright[height.length-1]=height[height.length-1];
for(int i=height.length-2;i>=0;i--)
maxright[i]=Math.max(maxright[i+1],height[i]);
int sum=0;
for(int i=0;i<height.length;i++){
int count=Math.min(maxright[i],maxleft[i])-height[i];
if(count>0) sum+=count;
}
return sum;
}
}
//单调栈
class Solution {
public int trap(int[] height) {
if(height.length<=2) return 0;
Stack<Integer> st=new Stack<>();
st.push(0);
int sum=0;
for(int i=1;i<height.length;i++){
if(height[i]<height[st.peek()]) st.push(i);
else if(height[i]==height[st.peek()]) {st.pop();st.push(i);}
else{
while(!st.isEmpty() && height[i]>height[st.peek()] ){
int mid=st.peek();
st.pop();
if(!st.isEmpty()){
int h=Math.min(height[i],height[st.peek()])-height[mid];
int w=i-st.peek()-1;
sum+=h*w;
}
}
st.push(i);
}
}
return sum;
}
}
总结
这道题有好多方法,可惜我一个都想不到。
每一列的雨水等于min(lHeight, rHeight) - height,能想明白这个,就能理解双指针和动态规划。
单调栈不太容易理解,模拟一遍代码运行过程,稍稍明白了点。但我觉得下次我还是写不出来吧。