leetcode 496. 下一个更大元素 I java题解

第二题：LeetCode 503. 下一个更大元素 II java题解

题目

https://leetcode-cn.com/problems/next-greater-element-i/

代码

nums1是nums2的子集，所以求nums2中每个元素的右边第一个比他大的元素。
计算nums2时用hashmap保存。

class Solution {
    public int[] nextGreaterElement(int[] nums1, int[] nums2) {
        int len1=nums1.length;
        int len2=nums2.length;
        LinkedList<Integer> stack=new LinkedList<>();
        HashMap<Integer,Integer> map=new HashMap<>();
        stack.addLast(nums2[len2-1]);
        for(int i=len2-2;i>=0;i--){
            int num=nums2[i];
            while(!stack.isEmpty()&&num>stack.peekLast()){
                stack.removeLast();
            }
            //找到栈里第一个比num大的元素
            //如果没有就是-1
            map.put(num,stack.isEmpty()?-1:stack.peekLast());
            stack.addLast(num);
        }
        for(int i=0;i<len1;i++){
            nums1[i]=map.getOrDefault(nums1[i],-1);
        }
        return nums1;
    }
}