给定一个整数数组 temperatures ,表示每天的温度,返回一个数组 answer ,其中 answer[i] 是指对于第 i 天,下一个更高温度出现在几天后。如果气温在这之后都不会升高,请在该位置用 0 来代替。
1.暴力解法超时了
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& temperatures) {
vector<int> answer(temperatures.size(),0);
for(int i=0;i<temperatures.size();i++){
for(int j=i;j<temperatures.size();j++){
if(temperatures[j]>temperatures[i]){
answer[i]=j-i;
break;
}
}
}
return answer;
}
};
2.单调栈
什么时候用单调栈:
通常是一维数组,要寻找任一元素的右边/左边第一个比自己大或自己小的元素位置。
单调栈的本质是空间换时间,因为在遍历的过程中需要用一个栈来记录右边第一个比当前元素的元素,优点是只需要遍历一次。
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& temperatures) {
vector<int> answer(temperatures.size(),0);
//单调栈
stack<int> st;
//栈里把下标压进去
st.push(0);
for(int i=1;i<temperatures.size();i++){
if(temperatures[i]<=temperatures[st.top()]) st.push(i);
else{
while(!st.empty()&&temperatures[i]>temperatures[st.top()]){
answer[st.top()]=i-st.top();
st.pop();
}
st.push(i);
}
}
return answer;
}
};