版权声明:本文为博主原创文章,博客地址:https://blog.csdn.net/qq_41855420,未经博主允许不得转载。 https://blog.csdn.net/qq_41855420/article/details/89761523
根据每日 气温 列表,请重新生成一个列表,对应位置的输入是你需要再等待多久温度才会升高的天数。如果之后都不会升高,请输入 0 来代替。
例如,给定一个列表 temperatures = [73, 74, 75, 71, 69, 72, 76, 73],你的输出应该是 [1, 1, 4, 2, 1, 1, 0, 0]。
提示:气温 列表长度的范围是 [1, 30000]。每个气温的值的都是 [30, 100] 范围内的整数。
思路分析: 和 上一道题 LeetCode 下一个更大元素I(hash表、栈) 一模一样,我们只要扫描数组的时候维护一个递减的栈,每次往栈顶放元素时,先与当前栈顶比较,如果栈顶元素小于当前元素,说明栈顶元素找到了下一个升高的温度。
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& T) {
stack<int> myStack;//递减栈,用于储存元素的下标
int Tsize = (int)T.size();
vector<int> resVec(Tsize, 0);
for (int index = 0; index < Tsize; ++index){//扫描整个数组
//当前栈顶的温度小于T[index],说明T[myStack.top()]找到了第一个升高的温度
while (!myStack.empty() && T[myStack.top()] < T[index]){
resVec[myStack.top()] = index - myStack.top();//计算天数差
myStack.pop();//已经找到所以出栈
}
myStack.push(index);//每次都需要放入栈中
}
return resVec;
}
};