题目描述
请根据每日气温列表 temperatures
,重新生成一个列表,要求其对应位置的输出为:要想观测到更高的气温,至少需要等待的天数。如果气温在这之后都不会升高,请在该位置用 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]
提示:
1 <= temperatures.length <= 105
30 <= temperatures[i] <= 100
解法一:暴力法
思路
由于温度范围在 [30, 100] 之内,并不是很大,因此可以维护一个数组 next
记录每个温度第一次出现时候的下标。由于我们要找的是至少要等多少天,所以数组 next
中的元素初始化为无穷大,并且在遍历温度列表的过程中更新 next
的值。
之后我们来反向遍历温度列表。对于每个元素 temperatures[i],在数组 next 中找到从 temperatures[i] + 1 到 100 中每个温度第一次出现的下标,将其中的最小下标记为 warmerIndex,则 warmerIndex 为下一次温度比当天高的下标。如果 warmerIndex 不为无穷大(即存在这天),则 warmerIndex - i 即为下一次温度比当天高的等待天数,最后再更新一下next数组的值,令 next[temperatures[i]] = i。
代码
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& temperatures) {
int n = temperatures.size();
vector<int> ans(n), next(101, INT_MAX);//初始化next数组
for (int i = n - 1; i >= 0; --i) { //寻找下次温度比当天高的下标
int warmerIndex = INT_MAX;
for (int t = temperatures[i] + 1; t <= 100; ++t) {
warmerIndex = min(warmerIndex, next[t]);
}
if (warmerIndex != INT_MAX) { //若找到了,更新答案数组
ans[i] = warmerIndex - i;
}
next[temperatures[i]] = i; //更新next数组
}
return ans;
}
};
就比如从temperatures
最后一个温度开始,进入循环,没发现目标下标,if判断不成立,便只更新了next数组
然后到倒数第二个温度,进入循环,若发现目标下标,if判断成立,更新next和ans数组
以此类推。
复杂度分析
- 时间复杂度:O(nm),其中 n 是温度列表的长度,m 是数组 next 的长度,在本题中温度不超过 100,所以 m 的值为 100。反向遍历温度列表一遍,对于温度列表中的每个值,都要遍历数组 next 一遍。
- 空间复杂度:O(m),其中 m 是数组 next 的长度。除了返回值以外,需要维护长度为 m 的数组 next 记录每个温度第一次出现的下标位置。
方法二:单调栈法
思路
可以维护一个存储下标的单调栈,从栈底到栈顶的下标对应的温度列表中的温度依次递减。如果一个下标在单调栈里,则表示尚未找到下一次温度更高的下标。
首先:正向遍历温度列表。对于温度列表中的每个元素 temperatures[ i ] ,如果栈为空,则直接将 i 进栈,如果栈不为空,则比较栈顶元素 top 对应的温度 temperatures[ top ] 和当前温度 temperatures[ i ] ,如果 temperatures[ i ] > temperatures[ top ] ,则将 top 移除(即找到了 top 对应的等待天数),并更新答案数组将 top 对应的等待天数赋为 i - top(记住这里的 i 和 top 是下标),重复上述操作直到栈为空或者栈顶元素对应的温度大于等于当前温度(即未找到下次温度更高的下标),然后将 i 进栈。
为什么可以放心的在弹栈的时候更新 ans[ top ] 呢?因为在这种情况下,即将进栈的 i 对应的 temperatures[ i ] 一定是 temperatures[ top ]及 栈中其他元素对应温度 右边第一个比它大的元素,试想如果 top 有 i 以外有其它大于等于它的元素,假设其下标为 j,那么 top 一定会在下标 j 的那一轮就被弹掉了。
由于单调栈满足从栈底到栈顶元素对应的温度递减,因此每次有元素进栈时,会将温度更低的元素全部移除,并更新出栈元素所对应的等待天数,这样便可以确保等待天数一定是最小的。
代码
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& temperatures) {
int n = temperatures.size();
vector<int> ans(n);
stack<int> s;
for (int i = 0; i < n; ++i) {
while (!s.empty() && temperatures[i] > temperatures[s.top()]) {
int top = s.top();
ans[top] = i - top;
s.pop();
}
s.push(i);
}
return ans;
}
};
优化
这里讲一个小技巧,刚学到的,如上代码可以再次进行时间方面的优化
具体如下:
当处理某些边界问题时,比如数组里面没数据了,这事需要做额外的判断,如果用一个guard,可以保持逻辑不变,统一判断逻辑,这样可以在时间方面进行一些优化
如以上代码的边界判断若改成:
s.push(INT_MAX);
......
......
while (s.top()!=INT_MAX && temperatures[i] > temperatures[s.top()])
则就:
有帮助记得点赞哇!