力扣42.接雨水

力扣42.接雨水

前后缀数组

  • 对于每个一个位置

    • 求其前面最高高度pre_max[i] = max(pre_max[i-1] , h[i])
    • 和后面最高高度suf_max[i] = max(suf_max[i+1] , h[i])
    • 当前i处的水容量 为min(pre_max[i] , suf_max[i]) - h[i]
cpp 复制代码
  class Solution {
  public:
      int trap(vector<int>& height) {
          int n = height.size();
          vector<int> pre_max(n),suf_max(n);
          pre_max[0] = height[0];
          suf_max[n-1] = height[n-1];
          for(int i=1;i<n;i++)
              pre_max[i] = max(pre_max[i-1],height[i]);
          for(int i=n-2;i>=0;i--)
              suf_max[i] = max(suf_max[i+1],height[i]);
          
          int res=0;
          for(int i=0;i<n;i++)
              res += min(pre_max[i],suf_max[i]) - height[i];
  
          return res;
      }
  };

双指针

  • 考虑不用数组存pre_max和suf_max而改用实时更新

    • 每到达一个位置时 更新pre_max和suf_max
    • 若pre_max < suf_max 说明当前水容量为 pre_max(小的) - h[i]
    • 反之亦然
cpp 复制代码
  class Solution {
  public:
      int trap(vector<int>& height) {
          int n = height.size();
          int l = 0,r = n-1;
          int res=0;
          int pre_max = 0,suf_max = 0;
          while(l <= r)
          {
              pre_max = max(pre_max,height[l]);
              suf_max = max(suf_max,height[r]);
              if(pre_max < suf_max)
              {
                  res += pre_max - height[l];
                  l ++;
              }
              else
              {
                  res += suf_max - height[r];
                  r --;
              }
          }
          return res;
      }
  };
相关推荐
Dream it possible!2 小时前
LeetCode 面试经典 150_矩阵_有效的数独(34_36_C++_中等)(额外数组)
leetcode·面试·矩阵
CoovallyAIHub2 小时前
4亿数据训练,零样本能力惊人:CLIP模型全解读
深度学习·算法·计算机视觉
Rotch2 小时前
核密度估计(KDE)(二)
算法
only-qi3 小时前
leetcode110. 平衡二叉树
数据结构·算法
要做朋鱼燕3 小时前
【C++】Vector核心实现:类设计到迭代器陷阱
开发语言·c++·笔记·算法·职场和发展
萘柰奈3 小时前
Unity学习----【数据持久化】二进制数据(五)--由Excel自动生成数据结构类与二进制文件
数据结构·学习·unity
Go_Zezhou4 小时前
在线性代数里聊聊word embedding
线性代数·算法·机器学习·nlp
Christo34 小时前
TFS-2005《A Possibilistic Fuzzy c-Means Clustering Algorithm》
人工智能·算法·机器学习
笨笨的摸索4 小时前
链表题类型注解解惑:理解Optional,理解ListNode
数据结构·经验分享·链表
hansang_IR4 小时前
【题解】洛谷P1776 宝物筛选 [单调队列优化多重背包]
c++·算法·动态规划·题解·背包·多重背包·单调队列