目录
- 1. 题目:
- 2. 我的代码:
- 小结:
1. 题目:
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 。
2. 我的代码:
class Solution:
def nextGreaterElement(self, nums1: List[int], nums2: List[int]) -> List[int]:
# 单调栈
stack = []
nums2_result = [-1] * len(nums2)
# 循环
stack.append(0)
for i in range(1, len(nums2)):
while stack != [] and nums2[i] > nums2[stack[-1]]:
k = stack.pop()
nums2_result[k] = nums2[i]
stack.append(i)
# 获取映射关系
relation = []
for i in nums1:
relation.append(nums2.index(i))
# 根据映射得到最终结果
return [nums2_result[k] for k in relation]
这个题其实只是多了个列表num1,我们需要做的就是还是按照第一个最大元素题目的求解规则,给num2求得第一个最大元素列表,然后截取出num1对应的部分,作为最终结果即可。
这里定义一个relation数组,表示映射关系(即:num1的索引位置对应的num2的索引位置)。第一个最大元素的单调栈就不多解释了,可以看:
每日温度
小结:
关注我给大家分享更多有趣的知识,以下是个人公众号,提供 ||代码兼职|| ||代码问题求解||
添加我的公众号即可: