给定一个整数数组 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]
这道题的关键在于用一个栈来存“未被解决的”下标。每遍历到一个新的温度,让它从栈顶开始解决,把能解决的都出栈(while 循环)。这其实会自然形成一个单调栈。
C++15 linesExpandCollapse
vector<int> dailyTemperatures(vector<int>& temperatures) {
int n = temperatures.size();
vector<int> ans(n, 0);
stack<int> stk;
for (int i = 0; i < n; i++) {
while (!stk.empty() && temperatures[i] > temperatures[stk.top()]) {
int day = stk.top();
stk.pop();
ans[day] = i - day;
}
stk.push(i);
}
return ans;
}