最初想到的解法是使用两层for循环进行遍历,找到每天的解,在提交时,但这个解法没有利用到题目给到的一些条件
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& temperatures) {
vector<int> answer(temperatures.size(), 0);
if(temperatures.size() == 1){
return {0};
}
for(size_t i = 0; i < temperatures.size() - 1; i++){
if(i >= 1 && temperatures[i] == temperatures[i - 1]){
if(answer[i - 1] > 0){
answer[i] = answer[i - 1] - 1;
}else {
answer[i] = answer[i - 1];
}
continue;
}
for(size_t j = i + 1; j < temperatures.size(); j++){
if(temperatures[i] < temperatures[j]){
answer[i] = j - i;
break;
}
}
}
return answer;
}
};
力扣的官方题解,暴力解法利用了30 <= temperatures[i] <= 100这个题目给定的条件,使用一个Next数组来记录每个温度出现的最早下标,然后从后往前来遍历 temperatures
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& temperatures) {
int n = temperatures.size();
vector<int> ans(n), next(101, INT_MAX);
for (int i = n - 1; i >= 0; --i) {
int warmerIndex = INT_MAX;
for (int t = temperatures[i] + 1; t <= 100; ++t) {
warmerIndex = min(warmerIndex, next[t]);
}
if (warmerIndex != INT_MAX) {
ans[i] = warmerIndex - i;
}
next[temperatures[i]] = i;
}
return ans;
}
};
以上两种方法都是使用双层for循环的暴力解法,在代码随想录学习到了单调栈的写法:
单调栈的使用场景:通常是一维数组,要寻找任一个元素的右边或者左边第一个比自己大或者小的元素的位置
单调栈的本质:是空间换时间,因为在遍历的过程中需要用一个栈来记录右边第一个比当前元素高的元素,优点是整个数组只需要遍历一次。
- 如果当前遍历的元素 大于栈顶元素,表示 栈顶元素的 右边的最大的元素就是 当前遍历的元素,所以弹出栈顶元素,并记录
- 如果栈不空的话,还要考虑新的栈顶与当前元素的大小关系,否则的话,可以直接入栈
- 注意,单调栈里 加入的元素是 下标。
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& temperatures) {
vector<int> answer(temperatures.size(), 0);
stack<int> st;
st.push(0);
for(int i = 1; i < temperatures.size(); i++){
if(temperatures[i] <= temperatures[st.top()]){
st.push(i);
}else{
while(!st.empty() && temperatures[i] > temperatures[st.top()]){
answer[st.top()] = i - st.top();
st.pop();
}
st.push(i);
}
}
return answer;
}
};