欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

496. 下一个更大元素 I

程序员文章站 2022-03-22 16:25:50
...

下一个更大元素 I

给定两个没有重复元素的数组 nums1 和 nums2 ,其中nums1 是 nums2 的子集。找到 nums1 中每个元素在 nums2 中的下一个比其大的值。

nums1 中数字 x 的下一个更大元素是指 x 在 nums2 中对应位置的右边的第一个比 x 大的元素。如果不存在,对应位置输出-1。

示例 1:

输入: nums1 = [4,1,2], nums2 = [1,3,4,2].
输出: [-1,3,-1]
解释:
    对于num1中的数字4,你无法在第二个数组中找到下一个更大的数字,因此输出 -1。
    对于num1中的数字1,第二个数组中数字1右边的下一个较大数字是 3。
    对于num1中的数字2,第二个数组中没有下一个更大的数字,因此输出 -1。

思路+代码+注释:

public int[] nextGreaterElement(int[] nums1, int[] nums2) {
        /*
            思路:使用map记录nums2中每个元素及其下标索引,然后遍历nums1中的元素在map中找到对应的下标,然后从该下标后开始查找比它大的元素,如果有进行记录,
            没有就记录-1
         */
        int[] res=new int[nums1.length];
        HashMap<Integer,Integer> map=new HashMap<>();
        for (int i = 0; i < nums2.length; i++) {
            map.put(nums2[i],i);
        }
        for (int i = 0; i < nums1.length; i++) {
            boolean flag=true;
            int num=nums1[i];
            int index=map.get(num);
            for (int j = index+1; j < nums2.length; j++) {
                if (nums2[j]>num)
                {
                    flag=false;
                    res[i]=nums2[j];
                    break;
                }
            }
            if (flag)
            {
                res[i]=-1;
            }
        }
        return res;
    }