LeetCode - 496 下一个更大元素 I-蒲公英云

LeetCode - 496 下一个更大元素 I

题目来源

题目描述

示例

提示

题目解析

算法源码

题目来源

496. 下一个更大元素 I - 力扣（LeetCode）

题目描述

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 = [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 。

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

提示

1 <= nums1.length <= nums2.length <= 1000
0 <= nums1[i], nums2[i] <= 104
nums1和nums2中所有整数互不相同
nums1 中的所有整数同样出现在 nums2 中

题目解析

此题和算法 - 找朋友_伏城之外的博客-CSDN博客

类似，用栈结构解题效率要比双重for更高。

另外，本题还有一个难点就是，本题只要输出nums1指定的数字对应的下一个更大元素，因此我们应该将nums2将每一个数字（定义为obj的key）对应的下一个元素（定义为obj.key的value）的，之后就可以简单地通过

nums1.map(ele => obj[ele]) 来得到题解。

算法源码

/**
 * @param {number[]} nums1
 * @param {number[]} nums2
 * @return {number[]}
 */
var nextGreaterElement = function (nums1, nums2) {
  let obj = {};
  let stack = [];
  for (let i = 0; i < nums2.length; i++) {
    let ele = nums2[i];
    obj[ele] = -1;
    while (true) {
      if (stack.length === 0) {
        stack.push(ele);
        break;
      } else {
        let peekEle = stack[stack.length - 1];
        if (ele > peekEle) {
          stack.pop();
          obj[peekEle] = ele;
        } else {
          stack.push(ele);
          break;
        }
      }
    }
  }
  return nums1.map((ele) => obj[ele]);
};