LeetCode 496 下乙個更大元素 I

2021-09-24 16:33:50 字數 1406 閱讀 9892

給定兩個沒有重複元素的陣列 nums1 和 nums2 ,其中nums1 是 nums2 的子集。找到 nums1 中每個元素在 nums2 中的下乙個比其大的值。

nums1 中數字 x 的下乙個更大元素是指 x 在 nums2 中對應位置的右邊的第乙個比 x 大的元素。如果不存在,對應位置輸出-1。

示例 1:

輸入: nums1 = [4,1,2], nums2 = [1,3,4,2].

輸出: [-1,3,-1]

解釋:對於num1中的數字4,你無法在第二個陣列中找到下乙個更大的數字,因此輸出 -1。

對於num1中的數字1,第二個陣列中數字1右邊的下乙個較大數字是 3。

對於num1中的數字2,第二個陣列中沒有下乙個更大的數字,因此輸出 -1。

示例 2:

輸入: nums1 = [2,4], nums2 = [1,2,3,4].

輸出: [3,-1]

解釋:對於num1中的數字2,第二個陣列中的下乙個較大數字是3。

對於num1中的數字4,第二個陣列中沒有下乙個更大的數字,因此輸出 -1。

注意:nums1和nums2中所有元素是唯一的。

nums1和nums2 的陣列大小都不超過1000。

思路:單調棧  維護乙個單調遞增的棧  棧中儲存元素的下標

單調棧用來解決「下乙個較好值「問題

class solution(object):

def nextgreaterelement(self, nums1, nums2):

""":type nums1: list[int]

:type nums2: list[int]

:rtype: list[int]

"""# 單調棧解決

d=self.help(nums2)

res=[d[num] for num in nums1]

return res

def help(self,nums):

'''找到陣列中的下乙個較大元素o(n) 單調棧模板

'''# 返回結果 key為當前數值 value為下乙個比它大的元素(下標)

res={}

# 棧用來儲存下標

stack=

# 維護乙個單調棧 此處為單調遞增棧 且從後往前遍歷

for i in range(len(nums)-1,-1,-1):

while stack and nums[stack[-1]]<=nums[i]:

stack.pop()

# 找到右邊第乙個較大的數 注意判斷棧是否為空

res[nums[i]]=nums[stack[-1]] if stack else -1

return res

Leetcode 496 下乙個更大元素

給定兩個沒有重複元素的陣列 nums1 和 nums2 其中nums1 是 nums2 的子集。找到 nums1 中每個元素在 nums2 中的下乙個比其大的值。nums1 中數字 x 的下乙個更大元素是指 x 在 nums2 中對應位置的右邊的第乙個比 x 大的元素。如果不存在,對應位置輸出 1。...

leetcode 496 下乙個更大元素 I

給定兩個沒有重複元素的陣列nums1和nums2,其中nums1是nums2的子集。找到nums1中每個元素在nums2中的下乙個比其大的值。nums1中數字 x 的下乙個更大元素是指 x 在nums2中對應位置的右邊的第乙個比 x 大的元素。如果不存在,對應位置輸出 1。示例 1 輸入 nums1...

leetcode 496 下乙個更大元素 I

給定兩個沒有重複元素的陣列nums1和nums2,其中nums1是nums2的子集。找到nums1中每個元素在nums2中的下乙個比其大的值。nums1中數字x的下乙個更大元素是指x在nums2中對應位置的右邊的第乙個比x大的元素。如果不存在,對應位置輸出 1。示例 1 輸入 nums1 4,1,2...