LeetCode–42.接雨水–双指针

题目:

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

分析:

这道题主要使用的是双指针方法,左右两个指针分别对应两边的柱子,并且同时向中间移动。

代码如下:

class Solution:
    def trap(self, height: List[int]) -> int:
        if not height:
            return 0

        n = len(height)
        left = 0
        right = n - 1
        maxleft = height[0]
        maxright = height[n - 1]

        ans = 0

        while left < right:
            maxleft = max(maxleft, height[left])
            maxright = max(maxright, height[right])
            if maxleft < maxright:
                ans += maxleft - height[left]
                left += 1
            else:
                ans += maxright - height[right]
                right -= 1
        return ans

你可能感兴趣的:(LeetCode–42.接雨水–双指针)