DAY 63单调栈day2

单调栈

503. 下一个更大元素 II

503. 下一个更大元素 II

题目描述

给定一个循环数组 nums （ nums[nums.length - 1] 的下一个元素是 nums[0] ），返回 nums 中每个元素的 下一个更大元素 。

数字 x 的 下一个更大的元素 是按数组遍历顺序，这个数字之后的第一个比它更大的数，这意味着你应该循环地搜索它的下一个更大的数。如果不存在，则输出 -1 。

思路

拼接两个nums数组，遍历，即可得到结果。
单调递增栈，① 栈顶元素>当前元素，当前元素入栈；
② 栈顶元素<当前元素，栈顶元素右侧第一个比他大的数就是当前元素，更新栈顶元素对应的结果，重复此过程，直至满足当前元素入栈结果。

更新结果是指定元素右侧第一个比他大的元素的数值。

代码

class Solution {
public:
    vector<int> nextGreaterElements(vector<int>& nums) {
        vector<int> nums1(nums.begin(), nums.end());
        nums.insert(nums.end(), nums1.begin(), nums1.end());
        vector<int> result(nums.size(), -1);
        stack<int> st;
        st.push(0);
        for (int i = 1; i < nums.size(); i++)
        {
            if (nums[i] <= nums[st.top()])
            {
                st.push(i);
            }
            else{
                while (!st.empty() && nums[i] > nums[st.top()])
                {
                    result[st.top()] = nums[i];
                    st.pop();
                }
                st.push(i);
            }
        }
        result.resize(nums.size() / 2);
        return result;
    }
};

42.接雨水

接雨水

题目描述

给定 n 个非负整数表示每个宽度为 1 的柱子的高度图，计算按此排列的柱子，下雨之后能接多少雨水。

思路

双指针
暴力解法及备忘录优化可参考: 暴力——备忘录优化——双指针
对于每一个元素，纵向的雨水高度取决于左右侧最高元素的较小一方，即res[i] = min(l_max, r_max) - height[i];
对于双指针方法，更新left指向元素的左侧最大值，更新right指向元素的右侧最大值。若对left元素，左侧最大值l_max小于右侧最大值r_max，说明该元素的高度取决于左侧高度，则更新ans，将left指针移动；同理，对于right指向的元素，如果右侧高度更低，说明左侧肯定有比这个高度更高的柱体，所以考虑r_max，更新结果并移动right指针即可。

代码

class Solution {
public:
    int trap(vector<int>& height) {
        int ans = 0;
        int n = height.size();
        int left, right;
        left = 0;
        right = n - 1;
        int l_max = height[0];
        int r_max = height[n - 1];
        while (left <= right)
        {
            l_max = max(l_max, height[left]);
            r_max = max(r_max, height[right]);

            if (l_max < r_max)
            {
                ans += l_max - height[left];
                left++;
            }
            else{
                ans += r_max - height[right];
                right--;
            }
        }
        return ans;
    }
};