代码随想录算法训练营第五十八天|● 739. 每日温度 ● 496.下一个更大元素 I

每日温度 

题目链接：739. 每日温度 - 力扣（LeetCode）

思路：很容易想到暴力解法。但超时也是很轻松的。

class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& temperatures) {
        // stack<int> dd;
        int dd = 1;
        vector<int> result(temperatures.size(), 0);
        if (temperatures.size() == 1)
            return {0};
        for (int i = 0; i < temperatures.size(); i++) {
            for (int j = i + 1; j < temperatures.size(); j++) {
                if (temperatures[j] > temperatures[i]) {
                    result[i] = dd;
                    break;
                } else {
                    dd++;
                }
            }
            dd=1;
        }
        return result;
    }
};

但是用栈解决，就不会超时了，我最开始没有用栈做出来，就是因为我存入栈的是数组内容，而不是下标。

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

 

下一个更大元素 I  

题目链接:496. 下一个更大元素 I - 力扣（LeetCode）

思路：单调栈还不是很熟练

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        stack<int> st;
        vector<int> result(nums1.size(), -1);
        if (nums1.size() == 0)
            return result;

        unordered_map<int, int> umap; // key:下标元素，value：下标
        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()]) {
                if (umap.count(nums2[st.top()]) >
                    0) { // 看map里是否存在这个元素
                    int index = umap[nums2[st.top()]]; // 根据map找到nums2[st.top()]
                                                       // 在 nums1中的下标
                    result[index] = nums2[i];
                }
                st.pop();
            }
            st.push(i);
        }
        return result;
    }
};