【力扣】下一个更大元素

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/next-greater-element-i

给定两个 没有重复元素 的数组 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。

class Solution:
    def nextGreaterElement(self, nums1: List[int], nums2: List[int]) -> List[int]:
        for idx,item in enumerate(nums1):
            y = nums2.index(item)
            for j in range(y,len(nums2)):
                if nums2[j] > item:
                    nums1[idx] = nums2[j]
                    break
                else:
                    nums1[idx] = -1
        return nums1

因为是暴力解法，这道题目我解的还是比较快的，需要注意两个点：

1）list列表返回索引方法的调用

2）在if判断语句找到符合条件的数值时，要及时break，不然还会继续遍历下去

可以看到我的这种思路表现是很不好的

 

 然后查看了一下人家的思路：递归栈+哈希表

• 时间复杂度：O(M+N)O(M+N)，其中N为nums1的长度，M为num2的长度
• 空间复杂度：O(M)O(M)，其中M为nums2的长度

class Solution:
    def nextGreaterElement(self, nums1: List[int], nums2: List[int]) -> List[int]:
        stack, hashmap = list(), dict()
        for i in nums2:
            while len(stack) != 0 and stack[-1] < i:hashmap[stack.pop()] = i
            stack.append(i)
        return [hashmap.get(i,-1) for i in nums1]

作者：qsctech-sange
链接：https://leetcode-cn.com/problems/next-greater-element-i/solution/python3-wu-xing-dai-ma-di-jian-zhan-ha-xi-biao-by-/
来源：力扣（LeetCode）

首先初始化一个列表和一个字典，我们假设nums2=[1,3,4,2]

循环1：不满足while条件，1进入stack，此时stack = [1]

循环2：字典hashmap[1] = 3 ,3进入stack，此时stack = [3]

循环3：字典hashmap[3] = 4 ,4进入stack，此时stack = [4]

循环4：不满足while条件，2进入stack，此时stack = [4,2]

建设nums1 = [4,1,2]

循环1：字典中无4，返回-1

循环2：字典中有1，返回1对应的键值3

循环3，字典中无2，返回-1

return [-1,3,-1]

hashmap起一个记录比当前元素大的右元素的作用

再举个例子来说，如果nums2里是3,2,1,4，那么栈前三位都是3,2,1，当遍历到4的时候，发现4比1大，这时候，哈希表里要添加hashmap[1] = 4,hashmap[2] = 4,hashmap[3] = 4。含义是，对于1这个数字而言，右边第一个比它大的数字是4。后面几个同理。