739. Daily Temperatures && 单调栈 && Python collections deque

题目大意

给你接下来每一天的气温，求出对于每一天的气温，下一次出现比它高气温的日期距现在要等多少天

 

解题思路

利用单调栈，维护一个单调递减的栈

将每一天的下标i入栈，维护一个温度递减的下标

若下一个温度p，比栈顶元素对应的温度p'要高，就出栈，且p就是p'的最近的“高温”

 

代码实现

from collections import deque
class Solution:
    def dailyTemperatures(self, temperatures):
        ans = [0]*len(temperatures)
        stack = deque()
        for i in range(len(temperatures)):
            if len(stack) == 0:
                stack.append(i)
            else:
                while temperatures[i] > temperatures[stack[len(stack)-1]]:
                    pos = stack.pop()
                    ans[pos] = i - pos
                    if len(stack) == 0:
                        break
                stack.append(i)
        return ans