代码随想录算法训练营第六十二天|503.下一个更大元素Ⅱ、42.接雨水

文档链接：https://programmercarl.com/

LeetCode503.下一个更大元素Ⅱ

题目链接：https://leetcode.cn/problems/next-greater-element-ii/

思路：乍一看跟下一个更大元素Ⅰ没什么不同，无非就是本题要循环数组了。

想法一：直接拼接一个nums在后面，遍历就可以了。

想法二：假装拼接一个，想法很巧妙。

单调栈：

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

---

LeetCode42.接雨水

题目链接：https://leetcode.cn/problems/trapping-rain-water/

思路：单调栈是按行来计算的。

暴力推荐按列计算。

暴力：

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 lHeight = height[i];
            int rHeight = height[i];
            for(int j = i + 1; j < height.size(); j++) {
                if(height[j] > rHeight) rHeight = height[j];
            }
            for(int j = i - 1; j >= 0; j--) {
                if(height[j] > lHeight) lHeight = height[j];
            }
            int h = min(lHeight, rHeight) - height[i];
            sum += h;
        }
        return sum;
    }
};

单调栈：

class Solution {
public:
    int trap(vector<int>& height) {
        stack<int> st;
        int sum = 0;
        st.push(0);
        for(int i = 1; i < height.size(); i++) {
            if(height[i] <= height[st.top()]) {
                st.push(i);
            } else {
                while(!st.empty() && height[i] > height[st.top()]) {
                    int mid = st.top(); 
                    st.pop();
                    if(!st.empty()) {
                        int h = min(height[st.top()], height[i]) - height[mid];
                        int w = i - st.top() - 1;
                        sum += h * w;
                    }
                }
                st.push(i);
            }
        }
        return sum;
    }
};

总结：补