题目:
给定一个整数数组 temperatures ,表示每天的温度,返回一个数组 answer ,其中 answer[i] 是指对于第 i 天,下一个更高温度出现在几天后。如果气温在这之后都不会升高,请在该位置用 0 来代替。
解析:
题目实现的目的是:求出下一个更大值与当前值的距离,因此只需要将值一个个的入进去,首先入第一个值,然后入第二个值,假如第二个值大于第一个值此时即找到了位置,此时就可以将第一个值出栈,假如第二个值小于等于第一个值,那么只有入栈操作,当遇到入的值大于栈顶的值,因为栈中的内容是递减的并不能保证所有的内容都小于要入的元素,所以此时要从后往前遍历栈中内容,一边比较栈顶元素和要入的元素一边出栈顶。最后将要入的元素入栈遍历完时此时就可以计算出结果了,栈内若还有值那么其与下一个更大值的距离为0。
总而言之栈中内容是递减的:将值一个个入栈,假如值大于栈顶,则找到了下一个更大值,此时要从后往前遍历栈中内容,一边比较栈顶元素和要入的元素一边出栈顶。最后将要入的元素入栈
代码:
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& temperatures) {
stack<int>st;
vector<int> result(temperatures.size(),0);
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()]){
result[st.top()] = i-st.top();
st.pop();
}
st.push(i);
}
}
return result;
}
};