即找之后第一个比自己大的元素
参考:
https://blog.youkuaiyun.com/u012836896/article/details/86657168
用栈实现:
将元素(下标)依次入栈并且保证栈内元素从栈顶到栈底是递增的。(如果当前元素比栈顶元素大,则说明当前元素就是比栈顶元素大的第一个元素,求出两个下标的差值即为栈顶元素对应的结果。依次出栈并计算下标差直到栈顶元素大于当前元素,将当前元素入栈)。
比如:T: [73, 74, 75, 71, 69, 72, 76, 73]
S:[73]
74大于73,说明比73大的第一个元素是74,两者下标差为1,即73处的结果为1 S:[74]
75大于74,说明比74大的第一个元素是75,下标差为1,74处的结果为1 S:[75]
71小于75,说明比75大的元素还没到,71入栈 S:[75,71]
69小于71,69入栈,S:[75,71,69]
72大于69,说明比69大的第一个元素是72,坐标差为1,69处的结果为1 S:[75,71]
同时72大于71,说明比71大的第一个元素是72,坐标差为2,71处的结果为2,S:[75]
72小于75,72入栈 S:[75,72]
76大于72,72处的结果为1,76大于75,75处的结果为4,栈空,76入栈,S:[76]
73小于76,入栈S:[76,73]
因此最终结果为[1,1,4,2,1,1,0,0]
代码:
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& T) {
if(T.empty())
return T;
int n = T.size();
vector<int> res(n,0);//记录结果,要初始化为0,因为最后如果没有元素比自己大,那么他的结果应该为0
stack<int> index;//保存元素下表
for(int i=0;i<n;i++){
while(!index.empty() && T[i]>T[index.top()]){//当前值大于栈顶元素就将栈顶元素出栈
res[index.top()] = i-index.top(); //栈顶元素的结果为当前元素下标和栈顶元素下标之差
index.pop(); //出栈
}
index.push(i); //直到当前元素大于栈顶元素,入栈
}
return res;
}
};