题目大意:根据每日气温列表,请重新生成一个列表,对应位置的输入是你需要再等待多久温度才会升高的天数。如果之后都不会升高,请输入 0 来代替。
即找到最近一个大于该元素的数的位置。
思路:求的是距离,那么就应该将下标入栈,最为合理。
代码如下:
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& t) {
vector<int> ret(t.size(),0);
stack<int> st;
for(int i=t.size()-1;i>=0;i--)
{
while(!st.empty() && t[i]>=t[st.top()])
{
st.pop();
}
ret[i]=st.empty()?0:st.top()-i;
st.push(i);
}
return ret;
}
};
解释:
st中,仅存有大于前一个数的节点,均为关键节点,节约了比较时间。比暴力双重循环解,性能更佳。
通过下标入栈,方便计算距离。