代码随想录算法训练营Day64|503.下一个更大元素II、42. 接雨水

目录

503.下一个更大元素II

前言

思路

方法一

方法二

42. 接雨水

前言

思路

算法实现

总结

---

503.下一个更大元素II

题目链接

文章链接

前言

         本题和“每日温度”类似，但是要求循环数组，利用单调栈实现就有些细微的差别了。

思路

        对于本题循环数组的处理，可以有两种策略：

        一种是将原数组扩充为原来的两倍，对扩充后的数组进行单调栈的处理，对最后求得的result数组进行压缩，仅保留原先一半大小的数据；

        第二种是不对原数组进行处理，而是在遍历顺序和循环中进行处理，在遍历的过程中模拟走了两遍原数组。

方法一

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;
    }
};

方法二

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()]) {
                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;
    }
};

42. 接雨水

题目链接

文章链接

前言

         本题在代码随想录中介绍了多种方法实现和优化处理，这里只介绍单调栈的方法实现。

思路

        单调栈是按照行方向来计算雨水，从栈头到栈底的顺序应该是从小到大的顺序。遇到相同的元素，更新栈内下标，就是将栈里元素（旧下标）弹出，将新元素（新下标）加入栈中。因为我们要求宽度的时候 如果遇到相同高度的柱子，需要使用最右边的柱子来计算宽度。

算法实现

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

总结

        明天就是训练营的最后一天了，坚持就是胜利！