代码随想录算法【Day47】

Day49

42. 接雨水

思路

这道题利用单调栈进行横向求解。对于每一个元素,找到它右边第一个比它大的元素和左边第一个比它大(或者与它相等的元素,当然这种情况可以忽略),最后计算雨水的存储量:((左右两边的高度差)减去(中间元素的高度))乘上中间元素的宽度。为什么要乘中间元素的宽度?因为中间元素可能不止一个。

代码

复制代码
class Solution {
public:
    int trap(vector<int>& height) {
        // if(height.size() <= 2) return 0;   //加不加都可以,最好加上
        stack<int> st;
        st.push(0);
        int sum;
        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()){   //如果这里是true,会发生什么?
                        int h = min(height[st.top()], height[i]) - height[mid];
                        int w = i - st.top() - 1;
                        sum += h * w;
                    }
                }
                st.push(i);
            }
        }
        return sum;
    }
};

等等

相关推荐
Codeking__44 分钟前
链表算法综合——重排链表
网络·算法·链表
minji...1 小时前
数据结构 堆(4)---TOP-K问题
java·数据结构·算法
AI_Keymaker2 小时前
一句话生成3D世界:腾讯开源混元3D模型
算法
Leon_vibs2 小时前
当 think 遇上 tool:深入解析 Agent 的规划之道
算法
旧时光巷2 小时前
【机器学习-2】 | 决策树算法基础/信息熵
算法·决策树·机器学习·id3算法·信息熵·c4.5算法
落了一地秋3 小时前
4.5 优化器中常见的梯度下降算法
人工智能·算法·机器学习
前端伪大叔3 小时前
第 5 篇:策略参数怎么调优?Freqtrade hyperopt 快速入门
算法·代码规范
Code季风3 小时前
深入理解令牌桶算法:实现分布式系统高效限流的秘籍
java·算法·微服务
KyollBM3 小时前
【Luogu】每日一题——Day15. P1144 最短路计数 (记忆化搜索 + 图论 + 最短路)
算法·图论
一百天成为python专家3 小时前
K-近邻算法
数据结构·python·算法·pandas·近邻算法·ipython·python3.11