题目:
给定一个整数数组 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
思路:单调栈
遍历整个数组,如果栈不空,且当前数字大于栈顶元素,那么如果直接入栈的话就不是 递减栈 ,所以需要取出栈顶元素,由于当前数字大于栈顶元素的数字,而且一定是第一个大于栈顶元素的数,直接求出下标差就是二者的距离。
继续看新的栈顶元素,直到当前数字小于等于栈顶元素停止,然后将数字入栈,这样就可以一直保持递减栈,且每个数字和第一个大于它的数的距离也可以算出来。
class Solution { public: vector<int> dailyTemperatures(vector<int>& temperatures) { int n = temperatures.size();//获取数组长度 vector<int>ss(n,0);//初始化答案数组 stack<int> st;//定义单调栈 for(int i=0;i<n;i++) { while(!st.empty()&&(temperatures[i]>temperatures[st.top()]))//当遇到进栈元素大于栈底元素且栈不为空时 { int t=st.top();//记录栈底元素下标 st.pop();//将栈底元素清出栈 ss[t]=i-t;//将答案记录进答案数组 } st.push(i);//记录栈顶元素下标 } return ss; } };