代码随想录刷题第58天|Leetcode739每日温度、Leetcode496下一个更大元素

1、Leetcode739每日温度

题目链接：739每日温度

第一种思路：暴力解法；

第二种思路：单调栈。

往栈中添加元素下标，第一个下标是0；然后从1开始遍历。

如果 t[1] <= t[0]，继续把 1 加入栈中；

如果 t[1] > t[0]，则要把0弹出把1入栈，同时result[0] = 1 - 0；

后续操作类似，如果 t[i] 小于等于 t[栈顶元素]，把 i 入栈；

如果 t[i] 大于 t[栈顶元素]，把栈顶元素弹出，result[st.top()] = i - st.top();

完成弹出和赋值的左右操作之后，栈中再加入新的下标索引。因为可能 t[i] 大于几个t[栈顶元素] 。

class Solution {
public:
    vector<int> dailyTemperatures(vector<int>& temperatures) {
        stack<int> st;
        vector<int> result(temperatures.size(), 0);
        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()])
                {
                    result[st.top()] = i - st.top();   
                    st.pop();                
                }
                st.push(i);
            }
        }

        return result;
    }
};

2、Leetcode496下一个更大元素

题目链接：496下一个更大元素

nums1 是 nums2的子集，找nums1中的元素在nums2中下一个比当前元素大的元素。

在遍历nums2的过程中，我们要判断nums2[i]是否在nums1中出现过，可以用map来做映射。

unordered_map<int, int> umap; // key:下标元素，value：下标
for (int i = 0; i < nums1.size(); i++) {
    umap[nums1[i]] = i;
}

后面就是用单调栈的思路。

遍历nums2，如果nums2[i] > nums2[st.top()]，再判断nums2[st.top()]是否在nums1数组里出现过。

如果出现过，即umap.count(nums2[st.top()]) > 0，result[umap[nums2[st.top()]]] = nums2[i];

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

        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++)
        {
            if (nums2[i] <= nums2[st.top()]) st.push(i);
            else
            {
                while (!st.empty() && nums2[i] > nums2[st.top()])
                {
                    if (umap.count(nums2[st.top()]) > 0)
                    {
                        result[umap[nums2[st.top()]]] = nums2[i];
                    }
                    st.pop();
                }
                st.push(i);
            }
        }

        return result;
    }
};