储存水..

问题描述:

给定m个非负整数表示每个宽度为1的柱子的高度图,计算按此排列的柱子下雨之后能储存多少水.

思路解析:

思考一下,什么样的位置能盛水?只有在当前柱子的左边和右边都比它高的情况下才能储存住水,而储水量和左侧最高柱及右侧最高柱有关.具体来说就是和左右两侧最矮的高度直接相关.变量如下:

height变量:表示输入的柱高度组

left变量:表示左指针指向的柱,初始值为0

right变量:表示右指针指向的柱,初始值为0

leftmax变量:表示左指针所指的左边最高的柱高

rightmax变量:表示右指针所指的柱右边最高的柱高

res变量:表示输出结果

完整代码如下:

def water(self, height):  # 定义一个名为water的方法，接收一个列表height作为参数
    left = 0  # 初始化左指针为0
    right = len(height) - 1  # 初始化右指针为列表的最后一个元素的索引
    leftmax = rightmax = 0  # 初始化左右两边的最大高度为0
    res = 0  # 初始化结果res为0，用于累积能够容纳的水的总量
    while left < right:  # 当左指针小于右指针时，继续循环
        if height[left] < height[right]:  # 如果左边的高度小于右边的高度
            if height[left] >= leftmax:  # 如果当前左边的高度大于等于左边的最大高度
                leftmax = height[left]  # 更新左边的最大高度
            else:  # 如果当前左边的高度小于左边的最大高度
                res += leftmax - height[left]  # 计算当前位置可以添加的水，并累加到res
            left += 1  # 移动左指针
        else:  # 如果左边的高度不小于右边的高度（即大于或等于）
            if height[right] >= rightmax:  # 如果当前右边的高度大于等于右边的最大高度
                rightmax = height[right]  # 更新右边的最大高度
            else:  # 如果当前右边的高度小于右边的最大高度
                res += rightmax - height[right]  # 计算当前位置可以添加的水，并累加到res
            right -= 1  # 移动右指针
    return res  # 返回最终计算出的水量