Day59 下一个更大元素Ⅱ + 接雨水

503 下一个更大元素Ⅱ

题目链接：503. 下一个更大元素 II - 力扣（LeetCode）

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

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

输入: nums = [1,2,1]
输出: [2,-1,2]
解释: 第一个 1 的下一个更大的数是 2；
数字 2 找不到下一个更大的数； 
第二个 1 的下一个最大的数需要循环搜索，结果也是 2。

思路：把两个数组拼接在一起，即可转换为上一题，但本题并不需要真正拼接，只需要遍历两次nums即可。

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

42 接雨水

题目链接：42. 接雨水 - 力扣（LeetCode）

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

输入：height = [0,1,0,2,1,0,1,3,2,1,2,1]
输出：6

思路：维护一个单调递减栈，在碰到递增时计算雨水，本方法雨水是按行计算的。

class Solution {
public:
    int trap(vector<int>& height) {
        int result = 0;
        stack<int> s; //递减栈
        s.push(0);
        for(int i = 1; i < height.size(); i++)
        {
            while(!s.empty() && height[i] > height[s.top()])
            {
                int a = s.top();
                s.pop();
                if(s.empty()) break;
                result += (i - s.top() - 1) * (min(height[s.top()], height[i]) - height[a]); 
            }  
            s.push(i);
        }
        return result;
    }
};