### 解题思路:利用栈来解决。
注意的是栈中压入的下标的值
首先压入0
如果栈不为空且要压入的下标值i有T[i]>T[栈顶元素] 则res[栈顶元素]=i-栈顶元素
否则继续压栈,直到压完T中所有的数据为止。
还要注意的是,开始对res全部赋值为0,这就对栈中的元素没弹出来的默认为0;
这里附一下图吧, 好理解一点。
代码如下:
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& T) {
int n = T.size();
//初始化vector数据,初始全部为0;
vector<int> res(n, 0);
stack<int> st;
for (int i = 0; i < n; ++i) {
//如果栈不为空,且选取的值大于下标为栈顶的值对应的数组值,则天数为下标之间的差
while (!st.empty() && T[i] > T[st.top()]) {
res[st.top()] = i - st.top();
st.pop();
}
//栈为空或者栈顶元素下标对应的值大于要压如栈的值
st.push(i);
}
return res;
}
};