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