leetcode 42 接雨水

最新推荐文章于 2024-09-09 22:01:06 发布

初学计算机的萌新

最新推荐文章于 2024-09-09 22:01:06 发布

阅读量98

点赞数

分类专栏： leetcode笔记文章标签： leetcode 算法

本文链接：https://blog.csdn.net/qq_24685471/article/details/116146977

版权

leetcode笔记专栏收录该内容

4 篇文章 0 订阅

订阅专栏

这篇博客介绍了LeetCode第42题‘接雨水’的两种解法。解法一是暴力求解，通过两层循环找出每个位置左右的最大高度来计算雨水量，时间复杂度为O(n^2)。解法二是利用韦恩图的思想，从两侧同时遍历数组，维护两个最大高度，快速求和并计算雨水量，时间复杂度为O(n)。这两种方法都要求空间复杂度为O(1)。

摘要由CSDN通过智能技术生成

leetcode 42 接雨水两种解法

第一次遇到这个题目是在数据结构课题目里面完全不会写，上leetcode找到了双指针你的写法，后来又找到此题回顾。

解法1 暴力解法

能存下雨水，说明雨水左右的高度都是大于雨水的高度的，因此我们只需要找到每个空位左右分别哪个最高就好了。

class Solution {
public:
    int trap(vector<int>& height) {
       int left_max = 0,right_max = 0;
       int sum = 0;
       int n = height.size();
       for(int i = 0;i<n-1;i++)
       {
           int left_max,right_max;
           left_max = 0;
           right_max = 0;
           for(int j = i;j>=0;j--)
           {
               if(height[j]>left_max)
               left_max = height[j];
           }
           for(int j = i;j<n;j++)
           {
               if(height[j]>right_max)
               right_max = height[j];
           }
           sum+= min(right_max,left_max)-height[i];
       }
    return sum;
    }
};

时间复杂度为O(n^2) 空间复杂度为O(1)

解法2 韦恩图法

class Solution {
public:
    int trap(vector<int>& height) {
        int n = height.size();
        int max1,max2,s1,s2;
        max1 = 0,max2 = 0;
        s1 = 0,s2 = 0;
        for(int i = 0;i<n;++i )
        {
            if(height[i]>max1)    
            max1 = height[i];
            if(height[n-i-1] > max2)
            max2 = height[n-i-1];
            s1+=max1;
            s2+=max2;
        }
        int sum = accumulate(height.begin(),height.end(),0); //快速求和
        return s1+s2 - n * max1 - sum;
    }
};