实现思想:
借助了【LeetCode84】-柱状图中最大矩形面积的思想,因为该计算也是满足后进先出的规律,所以选择用栈来实现,栈里面存储的是元素的下标值,只要当前的温度大于栈顶的元素,就计算两个下标值的差值,将该计算值作为栈顶元素下标对应的结果,同时将栈顶元素弹出,新的下标值压入栈中。这么处理完之后栈中还剩余的元素一律将计算值赋为0
实现代码:
#include <iostream>
#include <stack>
#include <vector>
using namespace std;
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& T) {
stack<int> s;
vector<int> re(T.size());
s.push(-1);
for (int i = 0;i < T.size();i++) {
while (s.top() != -1 && T[s.top()] < T[i]) {
int cur = s.top();s.pop();
re[cur] = i - cur;
}
s.push(i);
}
while (s.top()!=-1)
{
int cur = s.top();s.pop();
re[cur] = 0;
}
return re;
}
};
int main() {
return 0;
}
提交结果及分析:
时间复杂度和空间复杂度应该都是O(n)