Leetcode DAY 59: 下一个更大元素 and 接雨水

最新推荐文章于 2024-10-16 10:13:45 发布

Petal990912

最新推荐文章于 2024-10-16 10:13:45 发布

阅读量99

点赞数

文章标签： leetcode 数据结构算法

本文链接：https://blog.csdn.net/qq_44189622/article/details/129425565

版权

文章讲述了两个编程问题的解决方案，一个是503题，使用栈解决环状数组中找到每个元素的下一个更大元素；另一个是42题，接雨水问题，提供了双指针和单调栈两种方法计算存储的水量。在单调栈的实现中强调了忘记判断栈是否为空的错误。

摘要由CSDN通过智能技术生成

503.下一个更大元素II

相较于I，主要是数组变成了环状的，需要循环数组两边将 i下标替换为 i % nums.size()

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

42. 接雨水

1、双指针

用两个数组记录每个元素左边最大的元素和右边最大的元素

第i列的储水量为 min(Lh[i]， Rh[i]) - height[i]

用 Lh[i] = max(Lh[i - 1], height[i]) 对左边最大元素进行更新

class Solution {
public:
    int trap(vector<int>& height) {
        int n = height.size();
        vector<int> Lh(n, 0);  
        vector<int> Rh(n, 0);  
        int water = 0;
        stack<int> st;

        Lh[0] = height[0];
        for (int i = 1; i < n; i++) {
            Lh[i] = max(Lh[i - 1], height[i]);
        }

        Rh[n - 1] = height[n - 1];
        for (int i = n - 2; i >= 0; i--) {
            Rh[i] = max(Rh[i + 1], height[i]);
        }

        for (int i = 0; i < n; i++) {
            int count = min(Rh[i], Lh[i]) - height[i];
            if (count > 0) water += count;
        }
        return water;
    }
};

2、单调栈

主要是需要找到中间元素左边第一大的和右边第一大的

下标相减 - 1是宽 min(h左，h右) - h[mind] 是高

循环遍历height[i] 当height[i] 比栈顶元素大说明找到凹槽当前栈顶元素为mid 栈顶元素的下一个元素为左边最大当前i为右边最大

在实现过程中出现的问题：忘记在弹出栈顶元素之后判断栈是否为空

class Solution {
public:
    int trap(vector<int>& height) {
        int n = height.size();
        stack<int> st;
        st.push(0);
        int res = 0;
        for(int i = 1; i < n; 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 w = i - st.top() - 1;
                        int h = min(height[i], height[st.top()]) - height[mid];
                        res += w * h;
                    }
                }
                st.push(i);
            }
        }
        return res;
    }
};