Python算法题集_接雨水

本文为Python算法题集之一的代码示例

题目42:接雨水

说明:给定 n 个非负整数表示每个宽度为 1 的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水

示例 1:

复制代码
输入:height = [0,1,0,2,1,0,1,3,2,1,2,1]
输出:6
解释:上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图,在这种情况下,可以接 6 个单位的雨水(蓝色部分表示雨水)。 

注意:代码运行速度每次都不同,估计服务器负载有波动

  1. 分层双指针,差强人意

    ​      对图像进行分析,接雨水后高度为n+1的雨水一定在高度为n的雨水底座之上,类似金字塔;因此按高度分层,左右指针逐步向中间靠拢,最后得出雨水面积。此算法较为复杂,最终效果也差强人意。

    python 复制代码
    def trapRainWater_ext1(height):     # 分层双指针,按高度逐层上升
        ilen = len(height)
        ileft, iright, iSumbottom, iSumlevel, iLevel = 0, ilen - 1, 0, 0, 1
        while (ileft <= iright):
            while (ileft <= iright and height[ileft] < iLevel):
                iSumbottom += height[ileft]
                ileft += 1
            while (iright >= ileft and height[iright] < iLevel):
                iSumbottom += height[iright]
                iright -= 1
            iLevel += 1
            iSumlevel += iright - ileft + 1
        return iSumlevel - iSumbottom
    
    print(trapRainWater_ext1([0,1,0,2,1,0,1,3,2,1,2,1]))
    # 运行结果
    6
  2. 几何裁剪,数学之美,超越93%

基于几何图像分析,从左向右投射到最高峰,从右向左投射到最高峰,这两个面积相加,减去最高峰*宽的最高峰面积,就是装满雨水后的轮廓面积;这个轮廓面积再减去底座面积,就得出雨水占据的面积。

此算法简洁优雅,寥寥数行,速度居然超越93%的通过者

原来科学的尽头是玄学,美学的尽头是数学

python 复制代码
def trapRainWater_ext2(height): # 雨水面积=左边投射面积+右边投射面积-最高峰面积-底座面积
    result, hleft, hright = 0, 0, 0
    for iIdx in range(len(height)):
        hleft = max(hleft, height[iIdx])
        hright = max(hright, height[-iIdx - 1])
        result += hleft + hright - height[iIdx]
    return result - len(height) * hleft

print(trapRainWater_ext2([0,1,0,2,1,0,1,3,2,1,2,1]))
# 运行结果
6
  1. 双指针法,超越93%

    ​      抛弃高度分层的思路,直接使用左右指针相互靠拢;相当于去掉了一个中间层。轻装上阵后,效果也大大提高,代码虽然没有数学家优雅,效率也是超越了93%的通过者

    python 复制代码
    def traRainWater_ext3(height):  # 双指针收缩
        iLen = len(height)
        result, ileft, ileftMax, iright, irightMax= 0, 0, 0, iLen - 1, 0
        while ileft < iright:
            ileftMax = max(ileftMax, height[ileft])
            irightMax = max(irightMax, height[iright])
            if height[ileft] < height[iright]:
                result += ileftMax - height[ileft]
                ileft += 1
            else:
                result += irightMax - height[iright]
                iright -= 1
        return result
        
    print(trapRainWater_ext3([0,1,0,2,1,0,1,3,2,1,2,1]))
    # 运行结果
    6
  2. 堆栈大法超越97%

    ​      堆栈是编译原理中最常见的数据结构,采用堆栈来读取数组,精准分析雨水槽位置和面积,形成了降维打击。此算法超越97%的通过者,可谓是堆栈一出,谁与争锋

    python 复制代码
    def trapRainWater_ext4(height):     # 使用堆栈计算雨水槽
        stackDef = []
        res = 0
        for iIdx in range(len(height)):
            while stackDef and height[iIdx] > height[stackDef[-1]]:
                cur = stackDef.pop()
                if not stackDef:
                    break
                iHeight = min(height[iIdx], height[stackDef[-1]]) - height[cur]
                iWidth = iIdx - stackDef[-1] - 1
                res += iHeight * iWidth
            stackDef.append(iIdx)
        return res
    
        
    print(trapRainWater_ext4([0,1,0,2,1,0,1,3,2,1,2,1]))
    # 运行结果
    6

    一日练,一日功,一日不练十日空

    may the odds be ever in your favor ~

相关推荐
Young_Zn_Cu21 分钟前
LeetCode刷题记录(持续更新中)
算法·leetcode
天选之女wow26 分钟前
【代码随想录算法训练营——Day31】贪心算法——56.合并区间、738.单调递增的数字、968.监控二叉树
算法·leetcode·贪心算法
java1234_小锋28 分钟前
TensorFlow2 Python深度学习 - TensorFlow2框架入门 - 计算图和 tf.function 简介
python·深度学习·tensorflow·tensorflow2
lixinnnn.30 分钟前
贪心:火烧赤壁
数据结构·c++·算法
程序员晚枫31 分钟前
Python 3.14新特性:Zstandard压缩库正式加入标准库,性能提升30%
python
小小前端_我自坚强33 分钟前
前端算法相关详解
前端·算法
逆境清醒34 分钟前
VS Code配置Python开发环境系列(1)___VScode的安装 ,VScode常用快捷键
vscode·python·visual studio code
万粉变现经纪人1 小时前
如何解决 pip install -r requirements.txt 无效可编辑项 ‘e .‘(-e 拼写错误)问题
开发语言·python·r语言·beautifulsoup·pandas·pip·scipy
前端 贾公子1 小时前
《Vuejs设计与实现》第 5 章(非原始值响应式方案)下 Set 和 Map 的响应式代理
数据结构·算法
潇凝子潇1 小时前
在使用Nacos作为注册中心和配置中心时,如何解决服务发现延迟或配置更新不及时的问题
开发语言·python·服务发现