LeetCode-496. 下一个更大元素 I-Java-easy

题目链接

法一

    /**
     * （1）先遍历大数组nums2，首先将第一个元素入栈；
     * （2）继续遍历，当前元素小于栈顶元素时，继续将它入栈；当当前元素大于栈顶元素时，栈顶元素出栈，
     * （3）此时应将该出栈的元素与当前元素形成key-value键值对，存入HashMap中；
     * （4）当遍历完nums2后，得到nums2中元素所对应的下一个更大元素的hash表；
     * （5）遍历nums1的元素在hashMap中去查找‘下一个更大元素’，当找不到时则为-1。
     * @param nums1
     * @param nums2
     * @return
     */
    public int[] nextGreaterElement(int[] nums1, int[] nums2) {
        Deque<Integer> sk = new LinkedList<>();
        Map<Integer, Integer> mp = new HashMap<>();
        int[] ans = new int[nums1.length];
        for(int num : nums2){
            while(!sk.isEmpty() && sk.peek() < num){
                mp.put(sk.pop(), num);
            }
            sk.push(num);
        }
        for(int i = 0; i < nums1.length; i++){
            ans[i] = mp.getOrDefault(nums1[i], -1);
        }
        return ans;
    }

本地测试

        /**
         * 496. 下一个更大元素 I
         */
        lay.showTitle(496);
        Solution496 sol496 = new Solution496();
        int[] num496_1 = new int[]{4,1,2};
        int[] num496_2 = new int[]{1,3,4,2};
        int[] ans496 = sol496.nextGreaterElement(num496_1, num496_2);
        arrayOpt.showIntArray(ans496, ans496.length);