代码随想录算法训练营第五十八天丨739. 每日温度、496. 下一个更大元素 I

739. 每日温度

单调栈在前面单调队列的时候一并看过了，入门题自己写一遍。

class Solution:
    def dailyTemperatures(self, temperatures: List[int]) -> List[int]:
        ans = [0] * len(temperatures)
        mono = []
        for i, temperature in enumerate(temperatures):
            while mono and temperature > temperatures[mono[-1]]:
                ind = mono.pop()
                ans[ind] = i - ind
            mono.append(i)
        return ans

因为要找第一个比他大的，所以遍历到比栈顶元素大的目标的时候要弹出栈顶元素，因此维护一个（栈底到栈顶的）单调递减栈。

496. 下一个更大元素 I

单调栈就是干这个活儿的。

单调栈+哈希

class Solution:
    def nextGreaterElement(self, nums1: List[int], nums2: List[int]) -> List[int]:
        helper = collections.defaultdict(lambda : -1)
        mono = []
        for i, num in enumerate(nums2):
            while mono and num > mono[-1]:
                helper[mono.pop()] = num
            mono.append(num)
        ans = []
        for num in nums1:
            ans.append(helper[num])
        return ans

把代码随想录中的方法又拆开来写了，也因为是python才能这么写吧。

这里给一个代码随想录题解方法，一边维护单调栈一边填答案：

class Solution:
    def nextGreaterElement(self, nums1: List[int], nums2: List[int]) -> List[int]:
        ans = [-1] * len(nums1)
        helper = {num:i for i, num in enumerate(nums1)}
        mono = []
        for num in nums2:
            while mono and num > mono[-1]:
                if mono[-1] in helper:
                    ans[helper[mono[-1]]] = num
                mono.pop()
            mono.append(num)
        return ans

今日总结：

还是好理解的，第二题一开始题意没搞明白，没注意到nums1是nums2的子集，绕了好大圈子。

不得不说，python的defaultdict还是好用，可以大幅度简化代码...