leetcode 496. 下一个更大元素 I

输入:nums1 = [4,1,2], nums2 = [1,3,4,2].
输出:[-1,3,-1]
解释:nums1 中每个值的下一个更大元素如下所述:
- 4 ,用加粗斜体标识,nums2 = [1,3,4,2]。不存在下一个更大元素,所以答案是 -1 。
- 1 ,用加粗斜体标识,nums2 = [1,3,4,2]。下一个更大元素是 3 。
- 2 ,用加粗斜体标识,nums2 = [1,3,4,2]。不存在下一个更大元素,所以答案是 -1 。

解法:单调栈求一个数组中的下一个最大元素,用map保存对应元素。

class Solution {
    public int[] nextGreaterElement(int[] nums1, int[] nums2) {
        int len1=nums1.length,len2=nums2.length;
        Map<Integer,Integer> map=new HashMap<>();
        Deque<Integer> stack=new ArrayDeque<>();
        for(int i=0;i<len2;i++){
            while(!stack.isEmpty()&&nums2[i]>nums2[stack.peek()]){
                map.put(nums2[stack.peek()],nums2[i]);
                stack.pop();
            }
            stack.push(i);
        }
        int[] res=new int[len1];
        for(int i=0;i<len1;i++){
            res[i]=map.getOrDefault(nums1[i],-1);
        }
        return res;
    }
}
posted @ 2022-07-04 14:14  livingsu  阅读(24)  评论(0)    收藏  举报