ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

Leetcode 496 下一个更大元素 I

2022-02-21 09:34:00  阅读:135  来源: 互联网

标签:nums int 元素 nums1 textit 496 Leetcode nums2


题目描述

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

给你两个 没有重复元素 的数组 nums1nums2 ,下标从 0 开始计数,其中nums1nums2 的子集。

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

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

案例1:

输入: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 。

案例2:

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

方法一:暴力

我们可以暴力地逐个计算 \(\textit{nums}_1\)中的每个元素值 \(\textit{nums}_1[i]\) 在 \(\textit{nums}_2\)中对应位置的右边的第一个比 \(\textit{nums}_1[i]\) 大的元素值。具体地,我们使用如下方法:

  • 初始化与 \(\textit{nums}_1\) 等长的查询数组 \(\textit{res}\)。

  • 遍历 \(\textit{nums}_1\)中的所有元素,不妨设当前遍历到元素为 \(\textit{nums}_1[i]\):

    • 从前向后遍历 \(\textit{nums}_2\)中的元素,直至找到 \(\textit{nums}_2[j] = \textit{nums}_1[i]\)
    • 从 \(j+1\) 开始继续向后遍历,直至找到 \(\textit{nums}_2[k] > \textit{nums}_2[j]\)
    • 如果找到了 \(\textit{nums}_2[k]\),则将 \(\textit{res}[i]\) 置为 \(\textit{nums}_2[k]n\)],否则将 \(\textit{res}[i]\) 置为 \(−1\)。

    数组 \(\textit{res}\)即为最终结果。

    class Solution {
        public int[] nextGreaterElement(int[] nums1, int[] nums2) {
            int length = nums2.length;
            int[] ans = new int[nums1.length];
            for (int i = 0; i < nums1.length; i++) {
                int currNums1 = nums1[i];
                int j = 0;
                while (nums2[j]!=currNums1) {
                    j++;
                }
                int index = j + 1;
                while (index < length && nums2[index] <= currNums1) {
                    ++index;
                }
                if (index < length) {
                    ans[i] = nums2[index];
                } else {
                    ans[i] = -1;
                }
            }
            return ans;
        }
    }
    

方法二:单调栈

一旦要求下一个更大的元素,就是用单调栈解

我们可以先预处理 \(\textit{nums}_2\),使查询 \(\textit{nums}_1\)中的每个元素在 \(\textit{nums}_2n\) 中对应位置的右边的第一个更大的元素值时不需要再遍历 \(\textit{nums}_2\)。于是,我们将题目分解为两个子问题:

  • 第 \(1\) 个子问题:如何更高效地计算 \(\textit{nums}_2\) 中每个元素右边的第一个更大的值;
  • 第 \(2\) 个子问题:如何存储第 \(1\) 个子问题的结果。

我们可以使用单调栈来解决第 \(1\) 个子问题。倒序遍历 \(\textit{nums}_2\),并用单调栈中维护当前位置右边的更大的元素列表,从栈底到栈顶的元素是单调递减的。

具体地,每次我们移动到数组中一个新的位置 \(i\),就将当前单调栈中所有小于 \(\textit{nums}_2[i]\) 的元素弹出单调栈,当前位置右边的第一个更大的元素即为栈顶元素,如果栈为空则说明当前位置右边没有更大的元素。随后我们将位置 \(i\) 的元素入栈。

因为题目规定了 \(\textit{nums}_2n\)是没有重复元素的,所以我们可以使用哈希表来解决第 \(2\) 个子问题,将元素值与其右边第一个更大的元素值的对应关系存入哈希表。

class Solution {
    public int[] nextGreaterElement(int[] nums1, int[] nums2) {
        int length = nums2.length;
        int[] ans = new int[nums1.length];
        for (int i = 0; i < nums1.length; i++) {
            int currNums1 = nums1[i];
            int j = 0;
            while (nums2[j]!=currNums1) {
                j++;
            }
            int index = j + 1;
            while (index < length && nums2[index] <= currNums1) {
                ++index;
            }
            if (index < length) {
                ans[i] = nums2[index];
            } else {
                ans[i] = -1;
            }
        }
        return ans;
    }
}

标签:nums,int,元素,nums1,textit,496,Leetcode,nums2
来源: https://www.cnblogs.com/meiguiyulu/p/15917440.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有