LeetCode 739. 每日温度

LeetCode  ​​739. 每日温度​​

难度中等

请根据每日 ​​气温​​​ 列表，重新生成一个列表。对应位置的输出为：要想观测到更高的气温，至少需要等待的天数。如果气温在这之后都不会升高，请在该位置用 ​​0​​ 来代替。

class Solution:
    def dailyTemperatures(self, T: List[int]) -> List[int]:
        length = len(T)
        stack = []
        ans = [0] * length

        for i in range(length):
            while stack and T[i] > T[stack[-1]]:
                # 栈中有元素, 并且当前元素大于栈顶为索引的元素, 
                # 则将该索引弹出, 并且计算该索引和当前索引的距离作为结果
                pre_index = stack.pop()
                ans[pre_index] = i - pre_index
            stack.append(i)
        return ans

注意, stack中存储的是索引