维护一个单调递减(可以等于)的栈,栈内存的是元素的下标
当前元素小于等于栈顶元素时,直接入栈
否则,将栈顶元素出栈(对于栈顶元素,它找到了它的答案,即下标之差)
循环比较记录结果,直到当前元素入栈后,栈依然单调递减
- class Solution {
- public int[] dailyTemperatures(int[] temperatures) {
- int length = temperatures.length;
- int[] res = new int[length];
- Stack
stack = new Stack<>(); - stack.push(0);
- for(int i = 1; i < length; 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;
- }
- }
和前面差不多,只不过需要返回的结果要处理下
- class Solution {
- public int[] nextGreaterElement(int[] nums1, int[] nums2) {
- Map
map = new HashMap<>(); - for(int i = 0; i < nums1.length; i++){
- map.put(nums1[i], i);
- }
-
- Stack
stack = new Stack<>(); - stack.push(nums2[0]);
- int[] res = new int[nums1.length];
- Arrays.fill(res, -1);
- for(int i = 0; i < nums2.length; i++){
- if(nums2[i] <= stack.peek()){
- stack.push(nums2[i]);
- }else{
- while(!stack.isEmpty() && nums2[i] > stack.peek()){
- if(map.containsKey(stack.peek())){ // 只有当栈顶元素需要被记录时才记录
- int idx = map.get(stack.peek()); // 取出记录该在的位置
- res[idx] = nums2[i];
- }
- stack.pop(); // 不需要记录的元素也需要出栈,继续看前一个元素
- }
- stack.push(nums2[i]);
- }
- }
-
- return res;
- }
- }