解题记录 LeetCode 下一个更大元素 单调栈

题目链接: https://leetcode-cn.com/problems/next-greater-element-i/
题意: 给你两个 没有重复元素 的数组 nums1 和 nums2 ,其中nums1 是 nums2 的子集。
请你找出 nums1 中每个元素在 nums2 中的下一个比其大的值。
nums1 中数字 x 的下一个更大元素是指 x 在 nums2 中对应位置的右边的第一个比 x 大的元素。如果不存在,对应位置输出 -1 。
思路:
由于数组nums1是nums2的子集,整体思路是先求解出数组nums2中每个元素对应的下一个更大元素。然后存入map中,其中key是nums2中的元素值,value是下一个更大的元素。最后,遍历数组nums1,拿当前考察的元素作为key在map中获取value即可。
至此,关键问题是如何求解出数组nums2中每个元素对应的下一个更大元素。
我们逆序遍历数组nums2,对于当前考察的元素nums[i]来说:
如果栈不为空且nums[i]大于等于栈顶元素,说明栈中元素不会是其之前某个元素的下一个更大元素,所以将栈中元素出栈。原因在于,这里是逆序遍历,如果当前元素nums[i]大于栈中元素,那么对于其前面的元素来说,肯定nums[i]更可能是其下一个更大元素。
如果栈不为空且nums[i]小于栈顶元素,由于是逆序遍历,说明栈顶元素是nums[i]的下一个更大元素,所以将nums[i]作为key,栈顶元素作为value存入map。
如果栈为空,说明nums[i]没有下一个更大元素,所以将nums[i]作为key,-1作为value存入map。

最后,将nums[i]存入map,看其是否可能是之前某个元素的下一个更大元素。

class Solution {
    
    
    public int[] nextGreaterElement(int[] nums1, int[] nums2) {
    
    
        Map<Integer, Integer> map = new  HashMap<>();
        Deque<Integer> deque = new ArrayDeque<>();
        int ans[] = new int[nums1.length];
        for(int i = nums2.length - 1; i >= 0; i--) {
    
    
            while(!deque.isEmpty() && deque.peek() < nums2[i]) {
    
    
                deque.pop();
            }
            if(deque.isEmpty()) {
    
    
                map.put(nums2[i], -1);
            } else {
    
    
                map.put(nums2[i], deque.peek());
            }
            deque.push(nums2[i]);
        }
        for(int i = 0; i < nums1.length; i++) {
    
    
            ans[i] = map.get(nums1[i]);
        }
        return ans;
    }
}

Guess you like

Origin blog.csdn.net/qq_45560445/article/details/118486080