算法刷题记录 Day51

算法刷题记录 Day51

Date: 2024.04.19

lc 42. 接雨水

// 单调栈
class Solution {
public:
    int trap(vector<int>& height) {
        // 思路2：单调栈。当有个元素要入栈时。若该元素小于等于栈顶，则直接入栈；
        // 若该元素大于栈顶，则出栈。出栈后的新的栈顶，就是该出栈元素的左侧最大值。
        // (当前索引-新栈顶索引-1) * （min(左，右)-当前）， 即为累计的雨水;
        // 若出栈后无元素，则该元素自身为最大值，不统计该出栈值接了多少雨水。
        int n = height.size();
        int cur_idx = 0;
        stack<int> st;  // 存放单调栈的索引；

        int sum = 0;
        while(cur_idx < n){
            while(!st.empty() && height[cur_idx] >= height[st.top()]){
                int cur_pop_idx = st.top();
                st.pop();
                if(st.empty()){ //若出栈一个值后栈空，则该值自身为最大值，不累加雨水
                    
                }
                else{
                    // cout<<(min(height[cur_idx], height[st.top()]) - height[cur_pop_idx]) * (cur_idx-st.top()-1))<<endl;
                    // cout<<"当前索引:"<<cur_pop_idx<<endl;
                    // cout<<"前一索引:"<<st.top()<<endl;
                    // cout<<"后一索引:"<<st.top()<<endl;
                    // cout<<"当前累加值:("<<min(height[cur_idx], height[st.top()])<<" - "<<height[cur_pop_idx]<< ") * "<<(cur_idx-st.top()-1)<<endl;
                    sum += (min(height[cur_idx], height[st.top()]) - height[cur_pop_idx]) * (cur_idx-st.top()-1);
                }
            }
            st.push(cur_idx);
            cur_idx++;
        }
        return sum;
    }
};


// 双指针
class Solution {
public:
    int trap(vector<int>& height) {
        // 对于每个柱子（除头尾），能够接水的量是其左侧和右侧最大值中的较小值。
        // 因此，使用双指针，遍历两次数组，获取每个位置的左侧最大值和右侧最大值。
        // 再遍历一次数组，计算每个位置能接的雨水量并累计。
        int n = height.size();
        vector<int> leftHeight(n, 0);
        vector<int> rightHeight(n, 0);

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

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

        int sum = 0;
        for(int i=1; i<n-1; i++){
            sum += max(min(leftHeight[i], rightHeight[i]) - height[i], 0);
        }
        return sum;
    }
};

lc 503. 下一个更大元素II

class Solution {
public:
    vector<int> nextGreaterElements(vector<int>& nums) {
        // 相当于把两个数组相连后进行处理。
        vector<int> next(nums.size(), -1);
        int cur_idx = 0;
        stack<int> st;

        while(cur_idx < 2*nums.size()){
            int tmp_idx = cur_idx % nums.size();
            while(!st.empty() && nums[tmp_idx] > nums[st.top()]){
                if(next[st.top()] == -1){
                    next[st.top()] = nums[tmp_idx];
                }
                st.pop();
            }
            st.push(tmp_idx);
            cur_idx++;
        }
        return next;
    }
};