代码随想录算法训练营第五十八天

单调栈

找到数组（左边）右边第一个比他大（小）的元素是什么。

单调栈的作用

用来存放遍历过的元素，这样才能知道当前元素有没有比之前某个元素大。
存放方式，三种情况：
情况一：当前遍历的元素T[i]小于栈顶元素T[st.top()]的情况
情况二：当前遍历的元素T[i]等于栈顶元素T[st.top()]的情况
情况三：当前遍历的元素T[i]大于栈顶元素T[st.top()]的情况
再处理

739. 每日温度

题目链接：每日温度
流程：temperatures = [73,74,75,71,69,72,76,73]

• stack = [0], res = [0,0,0,0,0,0,0,0]
• 74>73, stack = [1], res = [1,0,0,0,0,0,0,0]
• 75>74, stack = [2], res = [1,1,0,0,0,0,0,0]
• 71<75, stack = [2,3], res = [1,1,0,0,0,0,0,0]
• 69<71, stack = [2,3,4], res = [1,1,0,0,0,0,0,0]
• 72>69, stack = [2,3], res = [1,1,0,0,1,0,0,0]
  stack = [2], res = [1,1,0,2,1,0,0,0]
  stack = [2,5], res = [1,1,0,2,1,0,0,0]
• 76>72, stack = [2], res = [1,1,0,2,1,1,0,0]
  stack = [6], res = [1,1,4,2,1,1,0,0]
• 73<76, stack = [6,7], res = [1,1,4,2,1,1,0,0]

class Solution:
    def dailyTemperatures(self, temperatures: List[int]) -> List[int]:
        stack = [0] #只放当前最大的元素的index
        res = [0]*len(temperatures)

        for i in range(1,len(temperatures)):
            if temperatures[i] <= temperatures[stack[-1]]:
                stack.append(i)
            else:
                while stack and temperatures[i] > temperatures[stack[-1]]:
                    # 每次都要更新，因为找的是附近最大的
                    res[stack[-1]] = i-stack[-1]
                    stack.pop()
                stack.append(i)
        return res

496.下一个更大元素 I

题目链接：下一个更大元素 I

class Solution:
    def nextGreaterElement(self, nums1: List[int], nums2: List[int]) -> List[int]:
        stack = [0]
        res = [-1]*len(nums1)
        
        for i in range(1,len(nums2)):
            if nums2[i] <= nums2[stack[-1]]:
                stack.append(i)
            else:
                while stack 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