每日一解 用栈就可以解决的每日温度

题目 每日温度

根据每日 气温 列表,请重新生成一个列表,对应位置的输出是需要再等待多久温度才会升高超过该日的天数。如果之后都不会升高,请在该位置用 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;
	}
};

就提交的结果来说,速度是很满足目标了,但是内存占用确实有点大,算是可以接受的空间换时间吧。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值