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]]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

 

你可能感兴趣的:(leetcode)