根据每日 气温
列表,请重新生成一个列表,对应位置的输入是你需要再等待多久温度才会升高的天数。如果之后都不会升高,请输入 0
来代替。
例如,给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73]
,你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]
。
提示:气温
列表长度的范围是 [1, 30000]
。每个气温的值的都是 [30, 100]
范围内的整数。
基本思路:维持一个递减栈。
遍历整个数组,判断一下栈顶元素和遍历到的数子的大小。若栈顶元素小于数字,那么把栈顶元素弹出,并找到栈顶元素在原数组中的位置,将Res数组中的该位置的值设置为(i-栈顶元素在原数组中的位置 其中i为循环中的变量值,Res数组就是我们要返回的结果。当然我们既要保存一个数的值也要保存一个数的下标,那么就得用一个Struct来保存啦) OK 再次进行比较数字和栈顶元素。直到栈为空或者是找到比数字大的数。
若栈顶小于数字,那么很简单 直接push到栈中即可。
AC解:
class Solution {
public:
struct Hash {
int Index;
int Value;
Hash() :Index(0), Value(0) {}
};
vector<int> dailyTemperatures(vector<int>& temperatures) {
int Len = temperatures.size();
vector<int> Res(Len, 0);
if (Len == 0) return Res;
Hash T; T.Index = 0; T.Value = temperatures[0];
stack<Hash> Temp;
Temp.push(T);
for (int i = 1; i<Len; i++) {
while (Temp.size() != 0 && temperatures[i]>Temp.top().Value) {
Res[Temp.top().Index] = i - Temp.top().Index;
Temp.pop();
}
Hash TT; TT.Index = i; TT.Value = temperatures[i];
Temp.push(TT);
}
return Res;
}
};