思路:可以用暴力搜索,时间复杂度为O(n2);或者使用递减栈:用栈存储每天温度的索引,如果新的一天的温度高于栈顶的就一直pop,并且存储索引的差保存到answer中,否则就push,注意栈中存储的是索引而非温度
原题链接:https://leetcode.cn/problems/daily-temperatures/description/?favorite=2cktkvj
1.题目如下:
给定一个整数数组 temperatures ,表示每天的温度,返回一个数组 answer ,其中 answer[i] 是指对于第 i 天,下一个更高温度出现在几天后。如果气温在这之后都不会升高,请在该位置用 0 来代替。
示例 1:
输入: temperatures = [73,74,75,71,69,72,76,73]
输出: [1,1,4,2,1,1,0,0]
示例 2:
输入: temperatures = [30,40,50,60]
输出: [1,1,1,0]
示例 3:
输入: temperatures = [30,60,90]
输出: [1,1,0]
提示:
1 <= temperatures.length <= 105
30 <= temperatures[i] <= 100
2.代码如下:
class Solution {
public:
//方案一:暴力搜索;时间复杂度O(n2)
//方案二:使用递减栈
/*
主要是用栈存储每天温度的索引,如果新的一天的温度高于栈顶的就一直pop,否则就push
注意栈中存储的是索引而非温度
*/
vector<int> dailyTemperatures(vector<int>& temperatures) {
int n=temperatures.size();
vector<int> ans(n,0);
stack<int> stackTemp;
for(int i=0;i<n;i++){
//注意这里一定要把判断栈是否为空放在前面,因为如果栈为空的话,则无法比较元素大小
while(!stackTemp.empty()&&temperatures[i]>temperatures[stackTemp.top()] ){
//栈存储的是每个元素的索引,而非元素值
auto idx=stackTemp.top();
stackTemp.pop();
ans[idx]=i-idx;
}
stackTemp.push(i);
}
return ans;
}
};