代码随想录算法训练营day58 | 739. 每日温度、496.下一个更大元素 I

739. 每日温度

使用单调栈方式，遇到当前元素比里面元素大的开始弹出记录结果，因为已经遇到温度更高的，遇到小的元素则直接存储，添加到栈内的元素一个比一个小。

class Solution:
    def dailyTemperatures(self, temperatures: List[int]) -> List[int]:
        result = [0] * len(temperatures)
        stack = []
        for i in range(len(temperatures)):
            while stack and temperatures[i] > temperatures[stack[-1]]:
                result[stack[-1]] = i - stack[-1]
                stack.pop()
            stack.append(i)
        return result

496.下一个更大元素 I 

相比于每日温度，主要是先将子集数组遍历一遍，存储在map中，方便遍历nums2的时候直接找到对应值的下标，写入结果。

class Solution:
    def nextGreaterElement(self, nums1: List[int], nums2: List[int]) -> List[int]:
        result = [-1] * len(nums1)
        stack = []

        nums1_dict = {}
        for i, num in enumerate(nums1):
            nums1_dict[num] = i

        for num in nums2:
            while stack and num > stack[-1]:
                if stack[-1] in nums1_dict:
                    result[nums1_dict[stack[-1]]] = num
                stack.pop()
            stack.append(num)
        return result