代码随想录（单调栈2）| 503.下一个更大元素II & 42. 接雨水

503.下一个更大元素II

这道题和 739. 每日温度 几乎如出一辙，可以自己尝试做一做
leetcode题目链接

文章讲解

本题就是一个环形的下一个元素

形成一个只有一圈的环的方法

循环的长度是vector长度的两倍，获取值的时候通过nums[i % nums.size()]

class Solution {
public:
    vector<int> nextGreaterElements(vector<int>& nums) {
        vector<int> result(nums.size(), -1);
        stack<int> stack;// 存储已经遍历过的数据的下标

        for(int i = 0; i < nums.size() * 2; i ++) {
            while(!stack.empty() && nums[i % nums.size()] > nums[stack.top()]) {
                // 每次都是给栈顶元素对应的位置赋值
                result[stack.top()] = nums[i % nums.size()];

                stack.pop();
            }
            //这里也要将 i取模后才能存储，因为stack存储的是数组下标，
            //如果这里不取模的话，在给result赋值的时候会越界
            stack.push(i % nums.size());
        }
        return result;
    }
};

42. 接雨水

leetcode题目链接
文章讲解

接雨水这道题目是 面试中特别高频的一道题，也是单调栈 应用的题目，大家好好做做。

建议是掌握 双指针 和单调栈，因为在面试中 写出单调栈可能 有点难度，但双指针思路更直接一些。

在时间紧张的情况有，能写出双指针法也是不错的，然后可以和面试官在慢慢讨论如何优化。

行和列的定义

按照行计算就是每次计算当前节点左右两侧第一个比当前节点高左右 两个节点所行成的区域

按照列的计算就是，每次计算当前节点以上到最高点的区域

单调栈是按照行方向来计算雨水

class Solution {
public:
    int trap(vector<int>& height) {
        int result = 0;
        stack<int> stack;

        for(int i = 0; i < height.size() ; i ++) {
            while(!stack.empty() && height[i] > height[stack.top()]) {
                int right_index = i;
                int cur_index = stack.top(); //每次就是在处理 栈顶元素产生的区域
                stack.pop();
                //如果当前节点比栈顶元素的值大， 那么 栈中的更深的节点就是
                if(!stack.empty()) {
                    int left_index = stack.top();
                    //求高度 差 左右比cur 大的值的最小值 减去cur的值 才是可以收集到的
                    int hight =  min(height[left_index], height[right_index]) - height[cur_index]; 
                    //求宽度 左边比他大的和右边比他大的 的下标差值 就是宽度
                    int wight = right_index - left_index - 1; //因为只求中间的差值所以要减一
                    result += hight * wight;

                }
            }
            stack.push(i);
        }
        return result;
    }
};

双指针按照列来结算

class Solution {
public:
    int trap(vector<int>& height) {
        int sum = 0;
        for (int i = 0; i < height.size(); i++) {
            // 第一个柱子和最后一个柱子不接雨水
            if (i == 0 || i == height.size() - 1) continue;

            int rHeight = height[i]; // 记录右边柱子的最高高度
            int lHeight = height[i]; // 记录左边柱子的最高高度
            for (int r = i + 1; r < height.size(); r++) {
                if (height[r] > rHeight) rHeight = height[r];
            }
            for (int l = i - 1; l >= 0; l--) {
                if (height[l] > lHeight) lHeight = height[l];
            }
            int h = min(lHeight, rHeight) - height[i];
            if (h > 0) sum += h;// 注意只有h大于零的时候，在统计到总和中
        }
        return sum;
    }
};