单调栈

问题:下一个更大元素

给定两个没有重复元素的数组nums1和nums2,其中nums1是nums2的子集。找到nums1中每个元素在nums2中的下一个比其大的值。
nums1 中数字 x 的下一个更大元素是指 x 在 nums2 中对应位置的右边的第一个比 x 大的元素。如果不存在,对应位置输出-1。

例如:输入: nums1 = [4,1,2], nums2 = [1,3,4,2].
输出: [-1,3,-1]

单调栈解法

维护一个单调递增的栈。
从右向左依此遍历nums2数组,当遍历的数x比栈顶元素小时,入栈(此时找到了第一个比x大的数,即为栈顶元素),并在map中记录。当遍历的数x比栈顶元素大时,将栈顶元素出栈,然后将x入栈,在x入栈之前需要判断栈是否为空,如果栈不为空,则x的第一个最大元素就是栈顶元素,如果栈为空,则x的下一个最大元素就是-1.

然后将nums1中的值作为key依此查找map,得到结果。

代码实现:

    public int[] nextGreaterElement(int[] nums1, int[] nums2) {
        int[] res = new int[nums1.length];
        Map<Integer,Integer> map = new HashMap<>();
        Stack<Integer> s = new Stack<>();
        for(int i=nums2.length-1;i>=0;i--){
            while(!s.isEmpty() && nums2[i] > s.peek()){ 
                s.pop();
            }
            if(s.isEmpty())
                map.put(nums2[i],-1);
            else{
                map.put(nums2[i],s.peek()); 
            }
            s.push(nums2[i]);
        }
        for(int i=0;i<nums1.length;i++){
            res[i] = map.get(nums1[i]);
        }
        return res;
    }
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值