每日温度
public int[] dailyTemperatures(int[] temperatures) {
int len=temperatures.length;
int []res=new int[len];
Deque<Integer> stack=new LinkedList<>();
stack.push(0);
for(int i=1;i<len;i++){
if(temperatures[i]<=temperatures[stack.peek()])stack.push(i);
else{
while(!stack.isEmpty()&&temperatures[i]>temperatures[stack.peek()]){
res[stack.peek()]=i-stack.peek();
stack.pop();
}
stack.push(i);
}
}
return res;
}
双端队列:Deque是一个接口,它的实现类有ArrayDeque和LinkedList。
当我们把Deque作为Stack使用时,注意只调用push()/pop()/peek()方法,不要调用addFirst()/removeFirst()/peekFirst()方法,那样会破坏栈的本质。
下一个更大元素
public int[] nextGreaterElement(int[] nums1, int[] nums2) {
int[]ans=new int[nums1.length];
Arrays.fill(ans,-1);
HashMap<Integer,Integer> map=new HashMap<>();
for(int i=0;i<nums1.length;i++){
map.put(nums1[i],i);
}
Deque<Integer> stack=new LinkedList<>();
for(int i=0;i<nums2.length;i++){
if(stack.isEmpty()){
stack.push(i);
}else{
while(!stack.isEmpty()&&nums2[stack.peek()]<nums2[i]){
int index=stack.pop();
if(map.containsKey(nums2[index])){
ans[map.get(nums2[index])]=nums2[i];
}
}
stack.push(i);
}
}
return ans;
}
下一个更大元素II
public int[] nextGreaterElements(int[] nums) {
Deque<Integer> stack=new LinkedList<>();
int []res=new int[nums.length];
Arrays.fill(res,-1);
for(int i=0;i<nums.length*2-1;i++){
int index=i%(nums.length);
if(stack.isEmpty()){
stack.push(index);
}else{
while(!stack.isEmpty()&&nums[stack.peek()]<nums[index]){
res[stack.peek()]=nums[index];
stack.pop();
}
stack.push(index);
}
}
return res;
}