【代码随想录】day58

weixin_470415552024-05-26 10:36

提示：文章写完后，目录可以自动生成，如何生成可参考右边的帮助文档

文章目录

[一、739. 每日温度](#一、739. 每日温度)
[二、496.下一个更大元素 I](#二、496.下一个更大元素 I)

一、739. 每日温度

暴力解超时了

cpp 复制代码

class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& temperatures) {
        int n = temperatures.size();
        vector<int> res(n, 0);
        for (int i = 0; i < n; i ++) {
            for (int j = i + 1; j < n; j ++) {
                if (temperatures[j] > temperatures[i]) {
                    res[i] = j - i;
                    break;
                }
            }
        }
        return res;      
    }
};

单调栈：适合求当前元素，左面或者右面第一个比当前元素大或者小的元素

cpp 复制代码

class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& temperatures) {
        int n = temperatures.size();
        vector<int> res(n, 0);
        stack<int> st;
        st.push(0);
        for (int i = 1; i < n; i ++) {
            while (!st.empty() && temperatures[i] > temperatures[st.top()]) {
                res[st.top()] = i - st.top();
                st.pop();
            }
            st.push(i);
        }
        return res;
    }
};

二、496.下一个更大元素 I

暴力搜索：

cpp 复制代码

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        vector<int> res(nums1.size(), -1);
        for (int i = 0; i < nums1.size(); i ++) {
            int tmp = -1;
            for (int j = nums2.size() - 1; j >= 0; j --) {
                if (nums2[j] > nums1[i]) {
                    res[i] = nums2[j];
                }
                if (nums2[j] == nums1[i]) {
                    break;
                }
            }
        }
        return res;
    }
};

单调栈：

cpp 复制代码

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        stack<int> st;
        vector<int> res(nums1.size(), -1);
        unordered_map<int, int> umap;
        for (int i = 0; i < nums1.size(); i ++) {
            umap[nums1[i]] = i;
        }
        st.push(0);
        for (int i = 1; i < nums2.size(); i ++) {
            while (!st.empty() && nums2[i] > nums2[st.top()]) {
                int num = nums2[st.top()];
                if (umap.count(num) > 0) {
                    res[umap[num]] = nums2[i];
                }
                st.pop();
            }
            st.push(i);
        }
        return res;
    }
};