给定一个整数数组 temperatures ,表示每天的温度,返回一个数组 answer ,其中 answer[i] 是指对于第 i 天,下一个更高温度出现在几天后。如果气温在这之后都不会升高,请在该位置用 0 来代替。
输入: temperatures = [73,74,75,71,69,72,76,73]
输出: [1,1,4,2,1,1,0,0]
思路:
1.暴力遍历两遍,算法时间复杂度为O(n^2)
2.要找每个元素右边第一个更大的元素,可以想到单调栈(
怎么能想到用单调栈呢? 什么时候用单调栈呢?
通常是一维数组,要寻找任一个元素的右边或者左边第一个比自己大或者小的元素的位置,此时我们就要想到可以用单调栈了。时间复杂度为O(n)。
)
在使用单调栈的时候首先要明确如下几点:
1.单调栈里存放的元素是什么?
单调栈里只需要存放元素的下标i就可以了,如果需要使用对应的元素,直接T[i]就可以获取。
此处代码我存放了用pair封装的值value和下标i
2.单调栈里元素是递增呢? 还是递减呢?
假设方向是从栈底到栈头(因为这也是入栈的顺序),当要找更大的元素,进一步推测新元素要小于栈头元素时要将新元素压入栈中,新元素要大于栈头元素时要将栈中元素pop一部分,直到空或更大元素出现。则应该采用从底向上的递减栈。
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& temperatures) {
stack<pair<int,int>> s;
vector<int> ret(temperatures.size(),0);
for(int i=0;i<temperatures.size();i++)
{
while(!s.empty()&&temperatures[i]>s.top().first)
{
auto temp=s.top();
s.pop();
ret[temp.second]=i-temp.second;
}
s.push(make_pair(temperatures[i],i));
}
while(!s.empty())
{
auto temp=s.top();
s.pop();
ret[temp.second]=0;
}
return ret;
}
};