根据每日 气温 列表,请重新生成一个列表,对应位置的输入是你需要再等待多久温度才会升高的天数。如果之后都不会升高,请输入 0 来代替。
例如,给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73],你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]。
提示:气温 列表长度的范围是 [1, 30000]。每个气温的值的都是 [30, 100] 范围内的整数。
凭借着我为数不多的编程经验,我觉得如果只要便利一遍气温数组,算法的效率就会很高。加上这道题提示可以使用栈做,于是就有了我下面的代码。
class Solution {
public:
/*
算法描述
如果栈不为空,拿到一个数,先和栈顶的元素比对。如果栈顶元素对应值小于该元素,栈顶元
素对应位置的ret等于当前元素的下标减当前元素索引,知道不满足条件位置,将当前值压入> 栈
当栈为空时,把当前数的下标压入栈中
*/
vector<int> dailyTemperatures(vector<int>& T) {
int T_size = T.size();
// 造出输出数组
vector<int> ret(T_size, 0);
// 暂存栈,如果遇到比栈顶元素大的数字,弹出
// 栈内保存的是元素下标
stack<int> myStack;
for(int i = 0; i != T_size; ++i) {
if(!myStack.empty()) {
while((!myStack.empty()) && (T[i] > T[myStack.top()])) {
ret[myStack.top()] = i - myStack.top();
myStack.pop();
}
}
myStack.push(i);
}
return ret;
}
};
不出所料,在所有的提交记录里果然是弟弟级别的。接下来学习一下第一的代码。
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& T) {
vector<int> res(T.size(),0);
for(int i=T.size()-2;i>=0;i--){
int n=i+1;
while(T[i]>=T[n]){
if(res[n]==0){
n=i;
break;
}else{
n+=res[n];
}
}
res[i]=n-i;
}
return res;
}
};
他的思路是,从倒数第二个温度开始算起(因为最后一个答案肯定是0)。i代表当前天,n代表比当前天温度大的天。
这里厉害的地方就是利用了输出数组的数学性质。n的起始天实在i的后一天,如果当前天气温比后一天的气温高,就会检查比这一天气温高的是哪一天。如果找到比当前天气温高的是哪一天,就会把n一到那一天的下标上;如果,找到了某一天,他的答案是0,而且当前天的气温还是比这一天的高,那么久说明在i以后不存在比当前天i气温更高的日子了,于是把n移到i,以方便res[i]=n-i;
计算输出数组。
利用了比大小的传递性,久没必要便利i后面的气温,以到达减少重复搜索的目的。答案是计算出来的,又没有入栈出栈的操作,自然比用栈实现要更好一点。