Leetcode 496. 下一个更大元素 I

庄园特聘拆椅狂魔

已于 2024-02-17 01:59:34 修改

阅读量322

点赞数 7

分类专栏：刷题训练营文章标签： leetcode 算法数据结构

于 2024-02-17 00:07:45 首次发布

本文链接：https://blog.csdn.net/lt_BeiMo/article/details/136133975

版权

刷题训练营专栏收录该内容

103 篇文章 3 订阅

订阅专栏

题意理解：

nums1 中数字 x 的 下一个更大元素 是指 x 在 nums2 中对应位置右侧的 第一个 比 x 大的元素。

        给你两个 没有重复元素 的数组 nums1 和 nums2 ，下标从 0 开始计数，其中nums1 是 nums2 的子集。

        对于每个 0 <= i < nums1.length ，找出满足 nums1[i] == nums2[j] 的下标 j ，并且在 nums2 确定 nums2[j] 的 下一个更大元素 。如果不存在下一个更大元素，那么本次查询的答案是 -1 。

        返回一个长度为 nums1.length 的数组 ans 作为答案，满足 ans[i] 是如上所述的 下一个更大元素 。

题意理解，实质上是指nums1中元素，在nums2中的位置上，其右边第一个比他大的值的下标。

其本质上也是一个单调栈问题，特别的是： nums1中元素和其下标设置映射关系，以获得nums1对应的result结果数组。

解题思路：

求一个元素其右边第一个比他大的元素。使用单调栈来解题。

其中单调栈维持单调增。

为了获得nums1 各个元素对应的reusult[i]值，需要维持nums1的元素和下标的映射关系，来组织result数组并返回。

map[nums1元素]=nums1元素下标

对nums2进行求解元素的右边第一个最大元素的下标

对其在nums1中的元素的对应result数组中赋值

1.单调栈解题

public int[] nextGreaterElement(int[] nums1, int[] nums2) {
        Map<Integer,Integer> map=new HashMap<>();
        for(int i=0;i<nums1.length;i++){
            map.put(nums1[i],i);
        }
        Stack<Integer> stack=new Stack<>();
        stack.push(0);
        int[] result=new int[nums1.length];
        Arrays.fill(result,-1);
        for(int i=1;i<nums2.length;i++){
            if(nums2[stack.peek()]>=nums2[i]){
                stack.push(i);
            }else{
                while((!stack.isEmpty())&&nums2[stack.peek()]<nums2[i]){
                    if(map.keySet().contains(nums2[stack.peek()])){
                        result[map.get(nums2[stack.peek()])]=nums2[i];
                    }
                    stack.pop();
                }
                stack.push(i);
            }
        }
        return result;
    }