题目

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

• 来源：力扣（LeetCode）
• 链接：leetcode.cn/problems/ne…
• 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解法一

思路

两边循环

代码

    public int[] nextGreaterElement(int[] nums1, int[] nums2) {
        int n1 = nums1.length;
        int n2 = nums2.length;
        int[] res = new int[n1];
        for (int i = 0; i < n1; i++) {
            int t = -1;
            boolean found = false;
            for (int j = 0; j < n2; j++) {
                if (nums2[j] == nums1[i]) {
                    found = true;
                }
                if (found && nums2[j] > nums1[i]) {
                    t = nums2[j];
                    break;
                }
            }
            res[i] = t;
        }
        return res;
    }

复杂度

• 时间复杂度：O(n^2)
• 空间复杂度：O(1)
• 上面的代码击败27%

解法二

思路

两个栈的方式

代码

    public int[] nextGreaterElement(int[] nums1, int[] nums2) {
        Stack<Integer> stack = new Stack<Integer>();
        Stack<Integer> tempStack = new Stack<Integer>();
        for (int num : nums2) {
            stack.push(num);
        }
        int n = nums1.length;
        int[] res = new int[n];
        for (int i = 0; i < n; i++) {
            int max = -1;
            while (!stack.isEmpty()) {
                int top = stack.peek();
                if (top == nums1[i]) {
                    res[i] = max;
                    break;
                } else if (top > nums1[i]) {
                    max = top;
                }
                tempStack.push(stack.pop());
            }
            while(!tempStack.isEmpty()) {
                stack.push(tempStack.pop());
            }
        }
        return res;
    }

复杂度

• 时间复杂度：O(mn)
• 空间复杂度：O(n)
• 这思路跑崩了。击败7%。

解法三

思路

• 单调栈

代码

    public int[] nextGreaterElement(int[] nums1, int[] nums2) {
        Stack<Integer> stack = new Stack<>();
        Map<Integer, Integer> map = new HashMap<>();
        for (int i = nums2.length - 1; i >= 0; i--) {
            int num = nums2[i];
            while (!stack.isEmpty() && num >= stack.peek()) {
                stack.pop();
            }
            int greater = stack.isEmpty() ? -1 : stack.peek();
            map.put(num, greater);
            stack.push(num);
        }
        int n = nums1.length;
        int[] res = new int[n];
        for (int i = 0; i < n; i++) {
            res[i] = map.get(nums1[i]);
        }
        return res;
    }

复杂度

• 时间复杂度：O(n)
• 空间复杂度：O(n)
• 上面代码击败了89%