北京市社会建设工作委员会网站大数据培训

题目

739. 每日温度

思路

创建一个栈，遍历输入，如果栈为空则直接压入，如果栈非空且当前温度大于栈顶元素，则弹出栈顶元素，并且a[pre]=i-pre，pre为栈顶元素，如果当前温度小于栈顶元素，则直接压入栈中。

代码

class Solution {
public:vector<int> dailyTemperatures(vector<int>& temperatures) {int n=temperatures.size();vector<int>a(n);stack<int>v;for(int i=0;i<n;i++){while(!v.empty() && temperatures[i]>temperatures[v.top()]){int pre=v.top();a[pre]=i-pre;v.pop();}v.push(i);}return a;}
};