No.739《每日温度》;题型:栈、哈希表
题目
题解思路(动态规划)
- 申请一个栈,用来存放元素下标;
- 遍历列表,若新元素小于栈顶元素,直接入栈;
- 若大于栈顶元素,则将其与栈顶元素的下标差存入结果数组的对应位置,然后弹出栈顶元素,并继续循环比较。
题解代码
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& T) {
int n=T.size();
vector<int>ans(n);
stack<int>s; //用来存放尚未找到更高温度的元素的下标
for(int i=0;i<n;i++){
while(!s.empty()&&T[i]>T[s.top()]){ //若新温度大于栈顶元素温度(有元素的话)
int x=s.top();
ans[x]=i-x; //将下标差值放入结果数组中(对应下标)
s.pop(); //栈顶元素弹出
}
s.push(i); //若新温度小于栈顶温度,直接入栈
}
return ans;
}
};
时间复杂度:O(n)
空间复杂度:O(n)
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/daily-temperatures
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。