leetcode:496. 下一个更大元素 I(单调栈)

链接:https://leetcode-cn.com/problems/next-greater-element-i/
找数组元素左/右第一个比它大的元素,都可以用单调栈解决。这道题跟第一个数组没什么关系,对第二个数组使用单调栈即可。
java代码:

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

python代码:

class Solution(object):
    def nextGreaterElement(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: List[int]
        """
        stack, hash = [],{}
        for n in nums2:
            while stack and stack[-1] < n:
                hash[stack.pop()] = n
            stack.append(n)
        return [hash.get(x,-1) for x in nums1]
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值