代码随想录算法训练营第53天（py）| 单调栈 | 739. 每日温度、496.下一个更大元素 I、503.下一个更大元素II

739. 每日温度

力扣链接
给定一个整数数组 temperatures ，表示每天的温度，返回一个数组 answer ，其中 answer[i] 是指对于第 i 天，下一个更高温度出现在几天后。如果气温在这之后都不会升高，请在该位置用 0 来代替。

思路1 暴力双循环

外层循环遍历每一个元素，内层循环遍历该元素后面的，找到第一个比他大的值。

class Solution:
    def dailyTemperatures(self, temperatures: List[int]) -> List[int]:
        answer = [0] * len(temperatures)
        for i in range( len(temperatures)): # i是当前温度
            for j in range(i+1, len(temperatures)): # j 是后面要找的温度
                if temperatures[j]>temperatures[i]: #如果遇到了更高温
                    answer[i] = j-i
                    break
        return answer

但是会超时

思路2 单调栈

一维数组中，要寻找任一元素的左/右第一个比自己大/小的元素的位置，就可以考虑单调栈。
单调栈的本质是空间换时间，就是用一个栈来记录我们遍历过的元素。

class Solution:
    def dailyTemperatures(self, temperatures: List[int]) -> List[int]:
        answer = [0] * len(temperatures)
        stack = [0] # 单调栈 用于存放下标 需要保持栈单调递减
        for i in range(1,len(temperatures)):
            if temperatures[i]<=temperatures[stack[-1]]:
                stack.append(i) 
                #如果当前元素比栈顶对应元素小，则压栈
            else:
                while len(stack)!=0 and temperatures[i]>temperatures[stack[-1]]:
                    answer[stack[-1]] = i - stack[-1]
                    stack.pop()
                stack.append(i)
                # 如果当前元素大，则把小的全部出栈并记录在结果中，再入栈当前元素
        return answer

496.下一个更大元素 I

力扣链接
给你两个 没有重复元素 的数组 nums1 和 nums2 ，其中nums1 是 nums2 的子集。

请你找出 nums1 中每个元素在 nums2 中的下一个比其大的值。

nums1 中数字 x 的下一个更大元素是指 x 在 nums2 中对应位置的右边的第一个比 x 大的元素。如果不存在，对应位置输出 -1 。

思路

class Solution:
    def nextGreaterElement(self, nums1: List[int], nums2: List[int]) -> List[int]:
        res = [-1]*len(nums1)
        stack = [0]
        for i in range(1,len(nums2)):
            if nums2[i] <= nums2[stack[-1]]:
                stack.append(i)
            else:
                while len(stack)!=0 and nums2[i]>nums2[stack[-1]]:
                    if nums2[stack[-1]] in nums1:
                        index = nums1.index(nums2[stack[-1]])
                        res[index] = nums2[i]
                    stack.pop()
                stack.append(i)
        return res

503.下一个更大元素II

力扣链接
给定一个循环数组，输出每个元素的下一个更大元素。

class Solution:
    def nextGreaterElements(self, nums: List[int]) -> List[int]:
        dp = [-1] * len(nums)
        stack = []
        for i in range(len(nums)*2):
            while(len(stack) != 0 and nums[i%len(nums)] > nums[stack[-1]]):
                    dp[stack[-1]] = nums[i%len(nums)]
                    stack.pop()
            stack.append(i%len(nums))
        return dp