leetcode第496题

leetcode第496题、下一个更大元素I

题目描述：
给定两个 没有重复元素 的数组 nums1 和 nums2 ，其中nums1 是 nums2 的子集。找到 nums1 中每个元素在 nums2 中的下一个比其大的值。
nums1 中数字 x 的下一个更大元素是指 x 在 nums2 中对应位置的右边的第一个比 x 大的元素。如果不存在，对应位置输出 -1 。

首先分析问题，寻找列表中下一个更大的元素的问题和栈的思想特别类似。首先定义一个栈stack和用来存放输出结果的ans。我的思路是先用栈的思路将数组二里的所有元素的结果排列好，然后将排列好的结果打包成一个哈希表，最后再遍历数组一将数组一的结果存放在res里。
注意，这里的栈存储的是数组二的元素索引，而不是元素值
1、遍历nums2，如果栈是空栈且该元素比栈顶的元素大，则pop出栈顶元素的索引，将改索引下的值存入ans中，通过这样一个循环我们可以得到一个nums2等长的ans
2、将nums2和ans用python的zip方法打包成字典dic
3、遍历nums1，输出对于字典的值
代码如下

def nextGreaterElement(nums1,nums2):
    lenth2 = len(nums2)
    ans = [-1]*lenth2
    lenth1 = len(nums1)
    res = [0]*lenth1
    stack = []
    for i in range(lenth2):
        num = nums2[i]
        while stack and num > nums2[stack[-1]]:
            s = stack.pop()
            ans[s] = nums2[i]
        stack.append(i)
    dic = dict(zip(nums2,ans))
    for i in range(lenth1):
        res[i] = dic[nums1[i]]
    return res