题目描述
参考文章:739. 每日温度
解题思路
(1)暴力法
每次遍历到一个数时,就再开辟一个变量找此数后面第一个大于它的数,找到则添加,没找到则返回0。
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& temperatures) {
int n = temperatures.size();
if(n == 1) return {0};
vector<int> res;
for(int i = 0; i < n; i++) {
int j = i + 1;
while(j < n && temperatures[i] >= temperatures[j]) {
j++;
}
if(j < n && temperatures[i] < temperatures[j]) {
res.push_back(j - i);
} else {
res.push_back(0);
}
}
return res;
}
};
(2)单调栈
单调栈的本质是空间换时间,因为在遍历的过程中需要用一个栈来记录右边第一个比当前元素高的元素,优点是整个数组只需要遍历一次。
更直白来说,就是用一个栈来记录我们遍历过的元素,因为我们遍历数组的时候,我们不知道之前都遍历了哪些元素,以至于遍历一个元素找不到是不是之前遍历过一个更小的,所以我们需要用一个容器(这里用单调栈)来记录我们遍历过的元素。
- 单调栈中存放元素:数组下标
- 单调栈中存放次序:从栈底到栈顶单调递减,当遇到比栈顶元素大时,弹栈;当遇到比栈顶元素小时压栈。此时可保证,该元素要为栈中若干个元素第一次遇到右边比它大的元素。
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& temperatures) {
int n = temperatures.size();
if(n == 1) return {0};
vector<int> res(n, 0);
stack<int> st, temp;
st.push(0);
for(int i = 1; i < n; i++) {
// 当遇到当天大于栈顶所存元素时,弹出栈顶元素,记录下标
while(!st.empty() && temperatures[st.top()] < temperatures[i]) {
res[st.top()] = i - st.top();
st.pop();
}
// 每次都将当天元素压入栈中
st.push(i);
}
return res;
}
};
参考文章:739. 每日温度