题目链接
题目大意
给定一个数组,对于每个元素a[i],求出下一个比该元素大的元素与该元素的距离。
class Solution
{
public:vector<int> dailyTemperatures(vector<int> &T){
int n = T.size();vector<int> res(n, 0);stack<int> sta;for (int i = n - 1; i >= 0; i--){
while (!sta.empty() && T[sta.top()] <= T[i])sta.pop();if (!sta.empty())res[i] = sta.top() - i;sta.push(i);}return res;}
};