作者:晓宜
🌈🌈🌈
个人简介:互联网大厂Java准入职,阿里云专家博主,csdn后端优质创作者,算法爱好者
🌙🌙🌙
分享单调栈的用法,有顺序和逆序两种解法,希望有所帮助
❤️❤️❤️
你的关注是我前进的动力😊
Problem: 739. 每日温度
文章目录
- 题目
- 思路
- 复杂度
题目
给定一个整数数组 temperatures ,表示每天的温度,返回一个数组 answer ,其中 answer[i] 是指对于第 i 天,下一个更高温度出现在几天后。如果气温在这之后都不会升高,请在该位置用 0 来代替。
示例 1:
输入: temperatures = [73,74,75,71,69,72,76,73]
输出: [1,1,4,2,1,1,0,0]
示例 2:
输入: temperatures = [30,40,50,60]
输出: [1,1,1,0]
示例 3:
输入: temperatures = [30,60,90]
输出: [1,1,0]
思路
我们有两种解法,分别是顺着遍历和倒着遍历,
先来看顺着遍历,我们从左到右遍历所有字符串,将他们加入到栈中;
我们要明白一点,就是当前栈顶元素是已经遍历过的的最近的元素,我们将它的值和此时遍历到的值比较,如果说当前栈顶元素小于现在的遍历的元素,对于当前栈顶元素而言,就找到了离他最近的点的元素,我们更新对应的ans值就可以了。
如果说当前元素小于等于栈顶元素,说明栈顶元素不需要更新,直接跳过就行
class Solution:def dailyTemperatures(self, temperatures: List[int]) -> List[int]:n = len(temperatures)ans = [0] * nstack = []for i in range(n):while stack!=[] and temperatures[stack[-1]] < temperatures[i]:index = stack[-1]stack.pop()ans[index] = i-indexstack.append(i)return ans
再来看看倒序遍历
倒序遍历相对而言就更加容易理解,我们本质是从后往前维护当前索引的左边的所有元素最大值如果栈顶元素小于当前元素,我们让他出栈,栈中只能保留大于当前元素的元素索
如果不为空,说明找到了最近的更大元素,更新答案即可,否则,前面就没有比当前元素更大的元素;最后将当前元素入栈
class Solution:def dailyTemperatures(self, temperatures: List[int]) -> List[int]:n = len(temperatures)ans = [0] * nstack = []for i in range(n-1,-1,-1):while stack and temperatures[stack[-1]]<=temperatures[i]:stack.pop()if stack:ans[i] = stack[-1] - istack.append(i)return ans
复杂度
时间复杂度:
O ( n ) O(n) O(n)
空间复杂度:
O ( n ) O(n) O(n)