leetcode739.每日温度「单调栈」

1.题目描述

根据每日 气温 列表，请重新生成一个列表，对应位置的输入是你需要再等待多久温度才会升高超过该日的天数。如果之后都不会升高，请在该位置用 0 来代替。

例如，给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73]，你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]。

提示：气温 列表长度的范围是 [1, 30000]。每个气温的值的均为华氏度，都是在 [30, 100] 范围内的整数。

2.解题思路

维护一个从大到小值递减的栈，但是栈中存放的是索引组成的列表。

遍历数组的每一个元素，

如果当前元素比栈顶（此时栈顶是由索引组成的列表，任意取一个索引，通常取0，对应的数值）大，那么弹出栈顶元素（由所有索引组成的列表）index_list，这个index_list 的每个index的离右侧最近且比它大的元素是当前元素的索引，因此结果res中index位置存入的是i-index

如果当前栈不为空且当前元素与栈顶列表中的第0个索引对应的值相等，则在栈顶的列表中压入这个元素的索引。（当有重复元素出现时，一定是要加入到栈顶的）

否则栈为空或者当前元素比栈顶列表索引对应的元素小，为当前这个元素的索引创建一个新的列表，并压入栈中。

遍历完所有元素后，如果此时栈不为空，则依次弹出每一个列表index_list，结果res中index位置存入的是0

解题思路和字节跳动面试题.找到数组中每一个元素左面和右面最近的比它小的值「单调栈」一样。

3.代码实现

class Solution(object):
    def dailyTemperatures(self, T):
        """
        :type T: List[int]
        :rtype: List[int]
        """
        size=len(T)
        res=[0 for _ in range(len(T))]
        stack = []
        for i in range(len(T)):
            while len(stack)>0 and T[stack[-1][0]]<T[i]:
                index_list=stack.pop(-1)
                for index in index_list:
                    res[index]=i-index
            if len(stack)>0 and T[stack[-1][0]]==T[i]:
                stack[-1].append(i)
            else:
                stack.append([i])
        while stack:
            index_list=stack.pop(-1)
            for index in index_list:
                res[index]=0
        return res

4.20210815更新

回顾此题，不知为何当初选取了那么繁琐的方式。。。

维护一个递减栈就好了，栈中存放的是原始列表的索引

遍历原始列表，当前值比栈顶索引对应值小或者相等，当前索引i直接入栈

当前值比栈顶索引对应值大，则弹出栈顶索引prevIndex，该索引对应的等待天数赋为 i - prevIndex

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