下一个最大元素
题目: 给定两个没有重复元素的数组,nums1和nums2,其中nums1是nums2的子集,找到nums1中每个元素在nums2中的下一个比其大的元素。
方法: 暴力法
思路:(1)字典dic中key为nums2中的元素,value为nums2中的每个元素的索引。
(2)遍历nums1中的元素。
(3)对于nums1中的元素,先找出其在nums2中的位置。
(4)遍历从此元素开始往后的每一个元素,输出第一个大于此元素的元素,若没有,则返回-1。
code
class Solution:
def nextGreaterElement(self, nums1: list[int], nums2: list[int]) -> list[int]:
hash_table = {} #注意:这里建立的是字典 键值对应,而不是列表
res = []
for i in range(len(nums2)):
hash_table[nums2[i]] = i
for num in nums1:
for i in range(hash_table[num] + 1, len(nums2)):
if nums2[i] > num:
res.append(nums2[i])
break
else:
res.append(-1)
return res
此题还有另一种方法:单调栈,等我掌握了再来更新!