DAY62 单调栈day1

DAY62

739. 每日温度

739. 每日温度

题目描述

给定一个整数数组 temperatures ，表示每天的温度，返回一个数组 answer ，其中 answer[i] 是指对于第 i 天，下一个更高温度出现在几天后。如果气温在这之后都不会升高，请在该位置用 0 来代替。

思路

对数组中每个元素，找右边第一个大于该元素的元素下标。
采用单调递增栈，栈顶元素到栈底元素递增。初始状态将数组的第一个元素入栈，此后可分为两种情况：

• 当前元素 <= 栈顶元素：当前元素入栈，栈中元素保持从栈顶向栈底单增。
• 当前元素 > 栈顶元素：说明找到了比栈顶元素大的第一个数。第一步，栈顶元素对应的的result结果可以更新。第二步，栈顶元素出栈，继续比较，直至满足入栈条件。

本题注意：

• result数组初始化为0——如果气温在这之后都不会升高，请在该位置用 0 来代替；
• 栈中存放下标即可，可通过下标取出元素；
• 栈顶元素出栈时，更新结果是栈顶元素下标与当前遍历元素的下标差值 i - st.top()。

代码

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;
    }
};

496.下一个更大元素 I

496.下一个更大元素 I

题目描述

nums1 中数字 x 的 下一个更大元素 是指 x 在 nums2 中对应位置 右侧 的 第一个 比 x 大的元素。
给你两个 没有重复元素 的数组 nums1 和 nums2 ，下标从 0 开始计数，其中nums1 是 nums2 的子集。
对于每个 0 <= i < nums1.length ，找出满足 nums1[i] == nums2[j] 的下标 j ，并且在 nums2 确定 nums2[j] 的 下一个更大元素 。如果不存在下一个更大元素，那么本次查询的答案是 -1 。
返回一个长度为 nums1.length 的数组 ans 作为答案，满足 ans[i] 是如上所述的 下一个更大元素 。

思路

可转换为大问题：求数组nums2中，当前元素右边第一个比该元素大的元素。
将nums1中元素加入unordered_map，对nums2处理时，判断是不是所求的nums1中有的元素，如果有，则进行求解，否则不必更新result。
栈st用于处理nums2，初始化时下标0入栈，对于当前元素：

• 当前元素 <= 栈顶元素：当前元素入栈；

• 当前元素 > 栈顶元素：栈顶元素找到了第一个位于其右侧的更大值。

• 1.判断该栈顶元素是否在所求的nums1中：

• 1.1若存在，则对nums1中该元素更新result；

• 1.2若不存在，忽略即可；

• 2.弹出栈顶元素

• 3.重复，至满足当前元素入栈条件，当前元素入栈。

本题注意：

• 题目提到了nums1和nums2无重复元素，可以用map；

• 使用单增栈，且result初始化为-1，大小为nums1大小——如果不存在下一个更大元素，那么本次查询的答案是 -1 。

• nums1中元素的result更新的是其右侧第一个大于该元素的数值。

代码

// 思路：先求出nums2的下一个更大元素，再根据nums1是nums2的子集，找到nums1中元素在nums2中下一个更大元素
// 思路：把nums1元素加入到umap，对于nums2，找到比元素
// 对于递增栈，弹出的时候是找到了第一个比栈顶元素大的元素，记录信息即可
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;
        for (int i = 0; i < nums1.size(); i++)
        {
            umap[nums1[i]] = i;
        }

        // 栈st用于处理nums2
        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.find(nums2[st.top()]) != umap.end())
                    {
                        int index = umap[nums2[st.top()]];
                        result[index]  = nums2[i];
                    }
                    st.pop();
                }
                st.push(i);
            }
        }
        return result;
    }
};