思路一:看到下一个更大->想到单调栈 O(n)
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& temperatures) {
//单调递减栈:左边大于等于 右边大于
int n = temperatures.size();
vector<int> ans(n, 0);
vector<int> stk(n);
int top = -1;
for (int i = 0; i < n; ++i) {
while (top != -1 && temperatures[stk[top]] < temperatures[i]) {
ans[stk[top]] = i - stk[top];
--top;
}
stk[++top] = i;
}
return ans;
}
};
自己感觉暴力做不来,实际可以做!!!
暴力:双循环O(n^7)
没想到的细节:内层对temperatures[i]的范围进行遍历即可,只有2次方,总共7次方,没问题!!!
class Solution {
public:
vector<int> dailyTemperatures(vector<int>& temperatures) {
vector<int> post(105, -1);
int n = temperatures.size(), max_ = temperatures[n - 1];
vector<int> ans(n, INT_MAX);
ans[n - 1] = 0;
post[temperatures[n - 1]] = n - 1; //别忘了
for (int i = n - 2; i >= 0; --i) {
for (int j = temperatures[i] + 1; j <= max_; ++j) {
if (post[j] != -1) {
ans[i] = min(ans[i], post[j] - i);
}
}
if (ans[i] == INT_MAX) ans[i] = 0;
max_ = max(max_, temperatures[i]);
post[temperatures[i]] = i;
}
return ans;
}
};
易错点:最后一个值要保存到post!!!
优化: 定义一个max_存i+1到n-1元素的最大值,内层循环遍历到max_就可以了。