题目 每日温度
根据每日 气温 列表,请重新生成一个列表,对应位置的输出是需要再等待多久温度才会升高超过该日的天数。如果之后都不会升高,请在该位置用 0 来代替。
例如,给定一个列表
temperatures = [73, 74, 75, 71, 69, 72, 76, 73]
你的输出应该是
[1, 1, 4, 2, 1, 1, 0, 0]
提示:气温 列表长度的范围是 [1, 30000]。每个气温的值的均为华氏度,都是在 [30, 100] 范围内的整数。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/daily-temperatures
思路
这里放一下我个人的思路过程:
看到这个题思考了几分钟,首先肯定想到的是暴力算法。但是列表长度范围达到了30000,如果采用一个O(n^2)的算法我丝毫不怀疑会超时。
然后根据经验想了想有没有可能上动态规划,一分析发现不满足最优子问题的条件,那也不是动态规划。
之后思考有没有可能使用外部排序。毕竟温度的范围题目已经给出了,只有71个数字,采用外部排序速度很快,而且符合温度值确定的要求。但是题目的要求是得到之后最近的哪一日比今日温度高,而不是得到正好高一度,也不是那么适合得到答案。
但是外部排序的比对思路和数据结构 给了我灵感。如果使用栈这个数据结构的话,先进后出,而且应当到这个题目中的话,也只需要O(n)的时间就可以解决问题。
举例,输入为示例输入:
[73, 74, 75, 71, 69, 72, 76, 73]
将73压入栈,注意还要保存73的位置0,也就是此时栈为:
[(73, 0)]
下一个数字为(74, 1),比此时栈末尾的73大,那么第0位的答案就是1 - 0 = 1,把(73, 0)抛出栈,同时把(74, 1)压入栈。
[(74, 1)]
下一个数字75和之前的74同理,第1位的答案是1,栈为:
[(75, 2)]
接下来数字71不比栈末尾的数字75大了,那么把这个71直接压进去就行:
[(75, 2), (71, 3)]
之后的69也是同样:
[(75, 2), (71, 3) ,(69, 4)]
下面的数字72比69大了,那么第4位的答案就是5 - 4 = 1,抛出69后,发现72也比71更大,那么第3位的答案就是5 - 3 = 2,然后72可不比75大,那么73压入栈:
[(75, 2), (72, 5) ]
基本上就是这样的思路,比末尾大就把末尾的数字得到答案并且抛出去,如果比末尾小或者栈为空就把当前的数字压进栈。
代码实现
思路算是简单而明确的思路了,那么代码实现自然也不算一件难事:
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& T) {
stack<int> daily_temperature;
stack<int> daily;
vector<int> answer;
for (int i = 0; i < T.size(); i++) {
answer.push_back(0);
if (daily.empty()) {
daily.push(i);
daily_temperature.push(T[i]);
continue;
}
while (!daily.empty() && daily_temperature.top() < T[i]) {
answer[daily.top()] = i - daily.top();
daily_temperature.pop();
daily.pop();
}
daily.push(i);
daily_temperature.push(T[i]);
}
return answer;
}
};
就提交的结果来说,速度是很满足目标了,但是内存占用确实有点大,算是可以接受的空间换时间吧。