739. 每日温度
1.题目
请根据每日 气温 列表,重新生成一个列表。对应位置的输出为:要想观测到更高的气温,至少需要等待的天数。如果气温在这之后都不会升高,请在该位置用 0 来代替。
2.实现
接触过单调队列,所以本题单调栈的方法呼之欲出
class Solution:
def dailyTemperatures(self, temperatures: List[int]) -> List[int]:
stack = []
n = len(temperatures)
res = [0] * n
stack.append(0)
for i in range(1, n):
if temperatures[stack[-1]] < temperatures[i]:
while stack and temperatures[stack[-1]] < temperatures[i]:
index = stack.pop()
res[index] = i - index
stack.append(i)
return res
3.单调栈介绍
1)使用场景:给定一维数组,要寻找任一个元素的右边或者左边第一个比自己大或者小的元素的位置
2)本质:空间换时间,因为在遍历的过程中需要用一个栈来记录右边第一个比当前元素高的元素,优点是只需要遍历一次。
3)方法:一般存放元素的下标i;比较栈顶和需入栈元素的大小来保持栈内递增还是递减的顺序
496. 下一个更大元素 I
1.题目
给你两个 没有重复元素 的数组 nums1 和 nums2 ,其中nums1 是 nums2 的子集。请你找出 nums1 中每个元素在 nums2 中的下一个比其大的值。
nums1 中数字 x 的下一个更大元素是指 x 在 nums2 中对应位置的右边的第一个比 x 大的元素。如果不存在,对应位置输出 -1 。
2.实现
困难点:这个题有点绕,对于找位置是没问题的,但是如何找到nums1对应数的值,没有想好怎么做
class Solution:
def nextGreaterElement(self, nums1: List[int], nums2: List[int]) -> List[int]:
stack = []
n = len(nums2)
stack.append(0)
m = len(nums1)
res = [-1] * m
a = defaultdict(int)
for i in range(m):
a[nums1[i]] = i
for i in range(n):
if nums2[stack[-1]] < nums2[i]:
while stack and nums2[stack[-1]] < nums2[i]:
index = stack.pop()
if nums2[index] in a:
res[a[nums2[index]]] = nums2[i]
stack.append(i)
return res
值得二刷