代码随想录训练营第54天|单调栈+双指针

42. 接雨水

cpp 复制代码
class Solution {
public:
    int trap(vector<int>& height) {
        int n=height.size(), sum=0;
        vector<int> max_left=height, max_right=height;
        for(int i=1; i<n; i++){
            max_left[i]=max(max_left[i-1],max_left[i]);
        }
        for(int i=n-2; i>=0; i--){
            max_right[i]=max(max_right[i],max_right[i+1]);
        }
        for(int i=0; i<n; i++){
            int cur=min(max_left[i],max_right[i])-height[i];
            if(cur>0)
                sum+=cur;
        }
        return sum;
    }
};

双指针解法:

max_left[i]:第i个元素以左(包含第i个),所有元素的最大值

max_right[i]:第i个元素以右(包含第i个),所有元素的最大值

按列统计雨水容量,每列可装的雨水高度:min(max_left[i],max_right[i])-height[i]

84. 柱状图中最大的矩形

cpp 复制代码
class Solution {
public:
    int largestRectangleArea(vector<int>& heights) {
        int n=heights.size(),max_area=INT_MIN;
        stack<int> st;
        vector<int> left(n,-1);
        vector<int> right(n,n);
        for(int i=0; i<n; i++){
            while(!st.empty()&&heights[i]<heights[st.top()]){
                right[st.top()]=i;
                st.pop();
            }
            st.push(i);
        }
        st=stack<int>();
        for(int i=n-1; i>=0; i--){
            while(!st.empty()&&heights[i]<heights[st.top()]){
                left[st.top()]=i;
                st.pop();
            }
            st.push(i);
        }
        for(int i=0; i<n; i++){
            int cur=heights[i]*(right[i]-left[i]+1-2);
            max_area=max(max_area,cur);
        }
        return max_area;

    }
};

尝试所有可能的中心高度,累计最大值。选定heights[i]作为中心,向两边膨胀得到最大面积,当遇到第一个比它小的元素时,则不可继续膨胀(否则会以更小的这个元素为中心,而不是heights[i]),也就是问题转化为求【下一个更小元素】的问题,参考上一篇博客,经典的单调栈解法。

从前往后扫,得到右侧第一个更小;

从后往前扫,得到左侧第一个更小。

注意,left和right的元素时取不到的,所以计算宽度时要减2。

相关推荐
Samuel-Gyx20 分钟前
数据结构--顺序表与链表
数据结构·算法·链表·1024程序员节
小年糕是糕手22 分钟前
【数据结构】队列“0”基础知识讲解 + 实战演练
c语言·开发语言·数据结构·c++·学习·算法
无限进步_26 分钟前
【C语言】函数指针数组:从条件分支到转移表的优雅进化
c语言·开发语言·数据结构·后端·算法·visual studio
Q741_1471 小时前
C++ 分治 快速选择算法 堆排序 TopK问题 力扣 215. 数组中的第K个最大元素 题解 每日一题
c++·算法·leetcode·分治·1024程序员节·topk问题·快速选择算法
文火冰糖的硅基工坊1 小时前
[人工智能-大模型-57]:模型层技术 - 软件开发的不同层面(如底层系统、中间件、应用层等),算法的类型、设计目标和实现方式存在显著差异。
人工智能·算法·中间件
鱼儿也有烦恼1 小时前
快速学完 LeetCode top 1~50 [特殊字符]
java·算法·leetcode·1024程序员节
独自破碎E1 小时前
LeetCode 380: O(1) 时间插入、删除和获取随机元素
java·算法·leetcode
Brookty1 小时前
【算法】前缀和(二)使用
java·学习·算法·前缀和·动态规划·1024程序员节
不良人龍木木1 小时前
云图-地基云图
图像处理·算法·计算机视觉·1024程序员节
RTC老炮1 小时前
webrtc弱网-PccNetworkController类源码分析与算法原理
网络·算法·webrtc