leetcode739.每日温度(中等)

leetcode739.每日温度(中等)_第1张图片
leetcode739.每日温度(中等)_第2张图片
思路一:看到下一个更大->想到单调栈 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_就可以了。

你可能感兴趣的:(#,栈&单调栈,leetcode739,每日温度)