用单调栈解决数组的下一个更大元素的问题

喝泉水的泉

于 2021-03-23 21:02:41 发布

阅读量94

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode

本文链接：https://blog.csdn.net/changbaishannefu/article/details/115143898

leetcode 专栏收录该内容

82 篇文章

订阅专栏

这段代码实现了一个名为`nextGreaterElement`的方法，该方法接收两个整数数组`nums1`和`nums2`，在`nums2`中寻找`nums1`中每个元素的下一个更大元素。它使用哈希映射存储`nums2`中的元素及其后续的更大元素，同时利用栈来处理当前元素和前一个元素的关系。最后将结果存储回`nums1`并返回。这种方法有效地解决了数组中查找下一个更大元素的问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在这里插入图片描述

class Solution {
    public int[] nextGreaterElement(int[] nums1, int[] nums2) {
        if(nums2.length==0) return null;
        HashMap<Integer, Integer> map = new HashMap<>();
        Stack<Integer> stack = new Stack<>();
        stack.push(nums2[0]);
        int i=1;
        while(i<nums2.length){
            
            while(!stack.empty()&&nums2[i]>stack.peek()) {
                 map.put(stack.peek(),nums2[i]);
                 stack.pop();
            }
            stack.push(nums2[i++]);
        }
        while(!stack.empty()){
            map.put(stack.pop(),-1);
        }

       for(int j=0;j<nums1.length;j++){
           nums1[j]=map.get(nums1[j]);
       }
       return nums1;


    }
}