北大也被挂，字节面试官有眼不识泰山。。。

精品推荐：

《征服数据结构》专栏：50多种数据结构彻底征服

《经典图论算法》专栏：50多种经典图论算法全部掌握

最近一北大算法工程师面试字节，结果挂了，网友评论：北大也被挂，字节面试官有眼不识泰山。在评论区还一个清华的算法工程师面试字节也被挂了，这说明算法工程师不但对学历有要求，对技能的要求也更高。

--------------下面是今天的算法题--------------

来看下今天的算法题，这题是LeetCode的第496题：下一个更大元素 I。

问题描述

来源：LeetCode第496题

难度：简单

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] 是如上所述的 下一个更大元素 。

示例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 。

• 1 <= nums1.length <= nums2.length <= 1000

• 0 <= nums1[i], nums2[i] <= 10^4

• nums1和nums2中所有整数互不相同

• nums1 中的所有整数同样出现在 nums2 中

问题分析

这题让计算nums1中每个元素在nums2 中对应位置右侧第一个比它大的值，其中nums1是nums2的子集，nums1和nums2中都没有重复的元素。实际上就是计算nums2中每一个元素右边第一个比它大的值，然后再使用nums1中的元素进行查找，我们可以使用单调栈来解决。

首先遍历nums2的所有元素，遍历的时候判断栈是否为空：

一，如果栈为空，就把遍历的元素加入到栈中。

二，如果栈不为空，查看栈顶元素是否小于当前元素：

     1，如果小于，说明栈顶元素遇到右边第一个比他大的值，然后栈顶元素出栈，记录下这个值。如果栈还不为空，继续比较……

     2，如果不小于，把当前遍历的元素加入到栈中。

JAVA：

public int[] nextGreaterElement(int[] nums1, int[] nums2) {
    // mp记录nums2中每个元素右边第一个比它大的值。
    Map<Integer, Integer> mp = new HashMap<>();
    // 单调栈，从栈顶到栈底是递增的
    Stack<Integer> stack = new Stack<>();
    // 遍历nums2的所有元素
    for (int num : nums2) {
        // 如果栈顶元素小于num，说明栈顶元素遇到了右边
        // 第一个比他大的值，然后栈顶元素出栈，记录下这个值。
        while (!stack.isEmpty() && stack.peek() < num)
            mp.put(stack.pop(), num);
        stack.push(num);// 当前元素入栈
    }
    // 因为nums1是nums2的子集，直接从mp中查找即可。
    int[] ans = new int[nums1.length];
    for (int i = 0; i < nums1.length; i++)
        ans[i] = mp.getOrDefault(nums1[i], -1);
    return ans;
}

C++：

public:
    vector<int> nextGreaterElement(vector<int> &nums1, vector<int> &nums2) {
        // mp记录nums2中每个元素右边第一个比它大的值。
        unordered_map<int, int> mp;
        // 单调栈，从栈顶到栈底是递增的
        stack<int> stk;
        // 遍历nums2的所有元素
        for (int num: nums2) {
            // 如果栈顶元素小于num，说明栈顶元素遇到了右边
            // 第一个比他大的值，然后栈顶元素出栈，记录下这个值。
            while (!stk.empty() && stk.top() < num) {
                mp[stk.top()] = num;
                stk.pop();
            }
            stk.push(num);// 当前元素入栈
        }
        // 因为nums1是nums2的子集，直接从mp中查找即可。
        vector<int> ans(nums1.size(), -1);
        for (int i = 0; i < nums1.size(); i++)
            if (mp.count(nums1[i]))
                ans[i] = mp[nums1[i]];
        return ans;
    }

Python：

def nextGreaterElement(self, nums1: List[int], nums2: List[int]) -> List[int]:
    # mp记录nums2中每个元素右边第一个比它大的值。
    mp = {}
    # 单调栈，从栈顶到栈底是递增的
    stack = []
    # 遍历nums2的所有元素
    for num in nums2:
        # 如果栈顶元素小于num，说明栈顶元素遇到了右边
        # 第一个比他大的值，然后栈顶元素出栈，记录下这个值。
        while stack and stack[-1] < num:
            mp[stack.pop()] = num
        stack.append(num)  # 当前元素入栈
    # 因为nums1是nums2的子集，直接从mp中查找即可。
    ans = [mp.get(num, -1) for num in nums1]
    return ans

笔者简介

博哥，真名：王一博，毕业十多年，《算法秘籍》作者，专注于数据结构和算法的讲解，在全球30多个算法网站中累计做题2000多道，在公众号中写算法题解800多题，对算法题有自己独特的解题思路和解题技巧，喜欢的可以给个关注，也可以下载我整理的1000多页的PDF算法文档。

《征服数据结构》专栏

数组，稀疏表(Sparse Table)，单向链表，双向链表，块状链表，跳表，队列和循环队列，双端队列，单调队列，栈，单调栈，双端栈，散列表，堆，字典树(Trie树)，ArrayMap，SparseArray，二叉树，二叉搜索树(BST)，笛卡尔树，AVL树，树堆(Treap)，FHQ-Treap

……

《经典图论算法》专栏

图的介绍，图的表示方式，邻接矩阵转换，广度优先搜索(BFS)，深度优先搜索(DFS)，A*搜索算法，迭代深化深度优先搜索(IDDFS)，IDA*算法，双向广度优先搜索，迪杰斯特拉算法(Dijkstra)，贝尔曼-福特算法(Bellman-Ford)，SPFA算法，弗洛伊德算法(Floyd)

……