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
209

被折叠的 条评论
为什么被折叠?



