写代码的第四十一天
开始单调栈了,以前没听过这个词,全新学习。。。
739. 每日温度
思路
根据题意可知,最后要的输出结果是找到第i天后面第一个比这一天温度高的第j天,然后输出j-i,i从0开始,代表当前数组的下标。所以根据输出我们要设置单调栈存储的是当前数组的下标,这样方便计算。分三种情况讨论,当第i天的温度小于当前栈顶元素的温度,那么将该元素入栈;当第i天的温度等于当前栈顶元素的温度,那么将该元素入栈;当第i天的温度大于当前栈顶元素的温度,那么查看当前温度的下标以及栈顶元素(栈顶元素代表的就是下标),做减法,然后将得到的数值append进result中,之后将栈顶元素弹出栈,将要插入的元素继续与新的栈顶元素比较,什么时候不大于栈顶元素了,什么时候停止,然后将要插入的元素插入栈,最后输出result。
错误第一版:result.append(i - stack[-1])这句话中没考虑清楚result存储的内容,result存储的是当前弹出的是stack栈顶元素,这个栈顶元素代表的是数组下标,所以在result中,也要找到对应的下标,在进行赋值。
class Solution:
def dailyTemperatures(self, temperatures: List[int]) -> List[int]:
result = []
stack = [0]
for i in range(len(temperatures)):
if temperatures[i] < temperatures[stack[-1]]:
stack.append(i)
elif temperatures[i] == temperatures[stack[-1]]:
stack.append(i)
else:
while len(stack) != 0 and temperatures[i] > temperatures[stack[-1]]:
result.append(i - stack[-1])
stack.pop()
stack.append(i)
return result
正确代码:
class Solution:
def dailyTemperatures(self, temperatures: List[int]) -> List[int]:
result = [0] * len(temperatures)
stack = [0]
for i in range(len(temperatures)):
if temperatures[i] < temperatures[stack[-1]]:
stack.append(i)
elif temperatures[i] == temperatures[stack[-1]]:
stack.append(i)
else:
while len(stack) != 0 and temperatures[i] > temperatures[stack[-1]]:
result[stack[-1]] = i - stack[-1]
stack.pop()
stack.append(i)
return result
496.下一个更大元素 I
思路
本题和上一题的区别在于,这是两个数组,我们需要知道两个数组之间的映射,根据nums1对nums2进行映射,找到对应的位置。所以本题的思路和上一题几乎一致。
第一步:定义输出数组result,他的长度和nums1的长度是一致的,因为我们是根据nums1去nums2找对应的,然后输出到和nums1下标位置对应的result数组中。
第二步:定义单调栈stack,栈存储的元素为数组下标。
第三步:对nums2进行单调栈操作,分三种情况讨论(这里的讨论和上一题是一样的)。1、当将要进入栈的数值小于等于栈顶元素对应的nums2时,将这个数值对应的nums2的下标入栈;2、当将要进入栈的数值大于栈顶元素对应的nums2时,将这个数值放入result中,此时这个数值对应的result下标为当前栈顶元素对应的nums1的数值的位置。
最后输出result。
错误第一版:错误原因ValueError: 3 is not in list
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
result[nums1.index(nums2[stack[-1]])] = nums2[i]
错误原因是,没考虑如果nums2有元素不在nums1中怎么办,所以应该加一个判断语句。
class Solution:
def nextGreaterElement(self, nums1: List[int], nums2: List[int]) -> List[int]:
result = [-1] * len(nums1)
stack = [0]
for i in range(len(nums2)):
if nums2[i] <= nums2[stack[-1]]:
stack.append(i)
else:
while len(stack) != 0 and nums2[i] > nums2[stack[-1]]:
result[nums1.index(nums2[stack[-1]])] = nums2[i]
stack.pop()
stack.append(i)
return result
正确代码:
class Solution:
def nextGreaterElement(self, nums1: List[int], nums2: List[int]) -> List[int]:
result = [-1] * len(nums1)
stack = [0]
for i in range(len(nums2)):
if nums2[i] <= nums2[stack[-1]]:
stack.append(i)
else:
while len(stack) != 0 and nums2[i] > nums2[stack[-1]]:
if nums2[stack[-1]] in nums1:
result[nums1.index(nums2[stack[-1]])] = nums2[i]
stack.pop()
stack.append(i)
return result
503.下一个更大元素II
思路
将nums数组和自身进行拼接!
正确代码:
class Solution:
def nextGreaterElements(self, nums: List[int]) -> List[int]:
# 拼接一个新的 nums
nums1 = nums.copy()
nums.extend(nums1)
# 用新的 nums 大小来初始化 result
result = [-1] * len(nums)
if not nums:
return result
# 开始单调栈
stack = [0]
for i in range(len(nums)):
if nums[i] <= nums[stack[-1]]:
stack.append(i)
else:
while stack and nums[i] > nums[stack[-1]]:
top = stack.pop()
result[top] = nums[i]
stack.append(i)
# 最后再把结果集即 result 数组 resize 到原数组大小
result = result[:len(nums)//2]
return result