1.题目描述
2.思路
(1)单调栈维护单调递增或者单调递减的数列
(2)因为要求找到当前元素 右边区域,第一个比当前元素大的元素,所以取单调增数量。
(3)单调栈存储元素的索引。如果遇到比当前遍历元素小的元素(或者等于当前元素的元素)直接加入到栈中。否则,开始比较当前遍历元素和栈顶元素的大小(要求当前遍历元素大于栈顶元素)
3.代码实现
class Solution {public int[] dailyTemperatures(int[] temperatures){int n=temperatures.length;//1.创建一个结果数组,默认初值是0int[] res=new int[n];//2.创建一个单调栈,用链表的形式,方便删除Deque<Integer> dq=new LinkedList<>();for(int i=0;i<n;i++){ //peekFirst() 虽然等价,但是双端队列专用术语,不如 peek() 直观表达“栈顶”。while(!dq.isEmpty()&&temperatures[i]>temperatures[dq.peek()]){int sum=i- dq.peek();int k=dq.pop();//赋值给 dq.peek() 弹出之前保存的索引位置。res[k]=sum;}dq.push(i);// 当前索引入栈}return res;}
}
带测试用例:
import java.util.Arrays;
import java.util.Deque;
import java.util.LinkedList;public class H739 {public int[] dailyTemperatures(int[] temperatures){int n=temperatures.length;//1.创建一个结果数组,默认初值是0int[] res=new int[n];//2.创建一个单调栈,用链表的形式,方便删除Deque<Integer> dq=new LinkedList<>();for(int i=0;i<n;i++){ //peekFirst() 虽然等价,但是双端队列专用术语,不如 peek() 直观表达“栈顶”。while(!dq.isEmpty()&&temperatures[i]>temperatures[dq.peek()]){int sum=i- dq.peek();int k=dq.pop();//赋值给 dq.peek() 弹出之前保存的索引位置。res[k]=sum;}dq.push(i);// 当前索引入栈}return res;}public static void main(String[] args){H739 test =new H739();int[] temperatures={73,74,75,71,69,72,76,73};int[] ans=test.dailyTemperatures(temperatures);System.out.println(Arrays.toString(ans));}
}