《LeetCode力扣练习》代码随想录——单调栈（下一个更大元素 I---Java）

最新推荐文章于 2024-06-29 10:15:27 发布

XRT_knives

最新推荐文章于 2024-06-29 10:15:27 发布

阅读量218

点赞数 2

分类专栏： LeetCode 文章标签： leetcode java 算法

本文链接：https://blog.csdn.net/XRT_knives/article/details/138282609

版权

LeetCode 专栏收录该内容

181 篇文章 2 订阅

订阅专栏

这篇文章介绍了如何使用单调栈算法解决LeetCode中的496号问题（下一个更大元素I），在Java中实现的思路和代码，通过HashMap和Deque数据结构来查找nums1中每个元素在nums2中的更大后继。

摘要由CSDN通过智能技术生成

                    
                    《LeetCode力扣练习》代码随想录——单调栈（下一个更大元素 I—Java） 
 
 刷题思路来源于 代码随想录 
 
496. 下一个更大元素 I 
 单调栈 class Solution {
    public int[] nextGreaterElement(int[] nums1, int[] nums2) {

        if (nums1.length == 0) {
            return new int[0];
        }

        int[] result = new int[nums1.length];
        Arrays.fill(result, -1);

        Map<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < nums1.length; i++) {
            map.put(nums1[i], i);
        }

        Deque<Integer> stack = new ArrayDeque<>();
        stack.push(0);

        for (int i = 1; i < nums2.length; i++) {

            if (nums2[i] <= nums2[stack.peek()]) {
                stack.push(i);
            } else {

                while (!stack.isEmpty() && nums2[i] > nums2[stack.peek()]) {

                    int index = nums2[stack.peek()];

                    if (map.containsKey(index)) {
                        result[map.get(index)] = nums2[i];
                    }

                    stack.pop();

                }

                stack.push(i);

            }

        }

        return result;

    }
}