题目
给定一个整数数组 temperatures ,表示每天的温度,返回一个数组 answer ,其中 answer[i] 是指在第 i 天之后,才会有更高的温度。如果气温在这之后都不会升高,请在该位置用 0 来代替。
方法1:双重循环。缺点:效率太低
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& temperatures) {
int num = temperatures.size();//根据数组temperatures的长度来创建数组。
vector<int> answer(num);//创建answer数组
for(int j=0;j<num;j++)
{
int after = 0;//记录几天后温度会升高
for(int i=j+1;i<num;i++)
{
if(temperatures[i]>temperatures[j])
{
answer[j] = after+1;
break;
}
else
{
after = after + 1;
}
}
}
answer[num-1] = 0;
return answer;
}
};
方法2:单调栈
- 单调栈中存放的是元素的下标
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& temperatures) {
int num = temperatures.size();//数组的长度
vector<int> answer(num); //创建结果数组
stack<int> st;//创建一个堆
st.push(0);
for(int i = 1; i < num;i++)
{
while(!st.empty() && temperatures[i] > temperatures[st.top()])//当下一个元素大于栈顶元素
{
answer[st.top()] = i - st.top();
st.pop();
}
st.push(i);
}
return answer;
}
};