496. 下一个更大元素 I

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

示例 1:

输入:nums1 = [4,1,2], nums2 = [1,3,4,2].
输出:[-1,3,-1]
解释:nums1 中每个值的下一个更大元素如下所述:
- 4 ,用加粗斜体标识,nums2 = [1,3,4,2]。不存在下一个更大元素,所以答案是 -1 。
- 1 ,用加粗斜体标识,nums2 = [1,3,4,2]。下一个更大元素是 3 。
- 2 ,用加粗斜体标识,nums2 = [1,3,4,2]。不存在下一个更大元素,所以答案是 -1 。

示例 2:

输入:nums1 = [2,4], nums2 = [1,2,3,4].
输出:[3,-1]
解释:nums1 中每个值的下一个更大元素如下所述:
- 2 ,用加粗斜体标识,nums2 = [1,2,3,4]。下一个更大元素是 3 。
- 4 ,用加粗斜体标识,nums2 = [1,2,3,4]。不存在下一个更大元素,所以答案是 -1 。

思路:暴力解法不用多说,对nums1的每个数,找到他在nums2中的下标,然后从该下标遍历到nums2末尾,找到第一个大于它的数,但这样的时间复杂度很大,感觉应该会超时。

那就利用上一篇博客中提到的单调栈来做!

首先result数组,长度和nums1一样,因为如果不存在是-1,因此初始化全置为-1.

然后对nums2数组用单调栈,因为要求第一个更大的元素,因此是递增的栈,若当前元素小于等于栈顶元素就入栈,否则遍历栈顶元素,若栈顶元素大于当前的元素,并且该栈顶元素在nums1中出现了,获取其在nums1中的下标index,然后更新result[index]的值为栈顶元素。 

代码(Python):

class Solution(object):
    def nextGreaterElement(self, nums1, nums2):
        result = [-1] *len(nums1)                    #结果数组,初始化全-1
        stack = [0]                                  #单调栈
        for i in range(1,len(nums2)):
            if nums2[i] <= nums2[stack[-1]]:         #若比栈顶元素小,入栈
                stack.append(i)
            else:
                while stack and nums2[i]>nums2[stack[-1]]:  #否则遍历大于当前元素的栈顶元素
                    if nums2[stack[-1]] in nums1:           #如果这个数在nums1出现了,更新result[index]
                        index = nums1.index(nums2[stack[-1]])
                        result[index] = nums2[i]
                    stack.pop()
                stack.append(i)
        return result
  • 2
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值