Day57单调栈 part01
739. 每日温度
leetcode:. - 力扣(LeetCode)
题意:请根据每日 气温 列表,重新生成一个列表。对应位置的输出为:要想观测到更高的气温,至少需要等待的天数。如果气温在这之后都不会升高,请在该位置用 0 来代替。
例如,给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73],你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]。提示:气温 列表长度的范围是 [1, 30000]。每个气温的值的均为华氏度,都是在 [30, 100] 范围内的整数。
思路:暴力解法是n^2的时间复杂度
什么时候用单调栈呢?通常是一维数组,要寻找任一个元素的右边或者左边第一个比自己大或者小的元素的位置,此时我们就要想到可以用单调栈了。时间复杂度为O(n)。单调栈就是保持这个栈内的元素是一直递增或者递减的。
- 单调栈里存放的元素是什么?
单调栈里只需要存放元素的下标i就可以了,如果需要使用对应的元素,直接T[i]就可以获取。
- 单调栈里元素是递增呢? 还是递减呢?
注意以下讲解中,顺序的描述为 从栈头到栈底的顺序,因为单纯的说从左到右或者从前到后,不说栈头朝哪个方向的话,大家一定比较懵。
即:如果求一个元素右边第一个更大元素,单调栈就是递增的,如果求一个元素右边第一个更小元素,单调栈就是递减的。
- 单调栈的作用和工作过程?记录一下遍历过的元素,和当前元素做对比
使用单调栈主要有三个判断条件。
- 当前遍历的元素T[i]小于栈顶元素T[st.top()]的情况:把T[i]压入栈
- 当前遍历的元素T[i]等于栈顶元素T[st.top()]的情况:把T[i]压入栈
- 当前遍历的元素T[i]大于栈顶元素T[st.top()]的情况:弹出st.top(), 把计算当前元素i-st.top()就是距离放入res[st.top()],把T[i]压入栈;如果下一个栈顶元素仍然小于T[i]则一直弹出指导是小于等于的情况
class Solution:
def dailyTemperatures(self, temperatures: List[int]) -> List[int]:
res = [0]*len(temperatures)
st = [0]
for i in range(1, len(temperatures)):
if temperatures[i] <= temperatures[st[-1]]:
st.append(i)
else:
while st and temperatures[i] > temperatures[st[-1]]:
pos = st.pop()
res[pos] = i-pos
st.append(i)
return res
496.下一个更大元素 I
leetcode链接:496. 下一个更大元素 I - 力扣(LeetCode)
题意:给你两个 没有重复元素 的数组 nums1 和 nums2 ,其中nums1 是 nums2 的子集。请你找出 nums1 中每个元素在 nums2 中的下一个比其大的值。nums1 中数字 x 的下一个更大元素是指 x 在 nums2 中对应位置的右边的第一个比 x 大的元素。如果不存在,对应位置输出 -1 。
示例 1:
输入: nums1 = [4,1,2], nums2 = [1,3,4,2]. 输出: [-1,3,-1] 解释: 对于 num1 中的数字 4 ,你无法在第二个数组中找到下一个更大的数字,因此输出 -1 。 对于 num1 中的数字 1 ,第二个数组中数字1右边的下一个较大数字是 3 。 对于 num1 中的数字 2 ,第二个数组中没有下一个更大的数字,因此输出 -1
思路:
这是一个递增栈,需要数组1和数组2之间有一个映射关系,然后数组2简历一个单调栈
class Solution:
def nextGreaterElement(self, nums1: List[int], nums2: List[int]) -> List[int]:
res = [-1]*len(nums1)
if len(nums1) == 0:
return res
st = [0]
for i in range(1, len(nums2)):
if nums2[i] <= nums2[st[-1]]:
st.append(i)
else:
while st and nums2[i] > nums2[st[-1]]:
if nums2[st[-1]] in nums1:
index = nums1.index(nums2[st[-1]]) #注意这个找寻index的做法
res[index] = nums2[i] #这是找元素,不是找距离了
st.pop()
st.append(i)
return res