python 水位_Leetcode 42. 接雨水 - python - 递归 查找分水岭

题目概括

黑色的是柱子，蓝色的是水。

求最多能盛多少水。

解法

有很多解法，比如扫描数组，算出每个位置上到左边、右边的最高柱子的高度， O(n)时间复杂度

下面的解法是找分水岭，比如从左往右遍历，只要找到第一个高度大于最左侧的柱子的，它就是分水岭，这时候就能确定前面的水位。然后后面的和前面的就没关系了，递归执行这个算法。

到最后一段，可能没有更高的柱子了，找不到分水岭，但是可以直接把数组前后颠倒(reverse)，再递归调用一次就可以了。

class Solution:

def trap(self, height: List[int]) -> int:

cnt = len(height)

if cnt < 3: return 0

if cnt == 3:

if height[0] > height[1] < height[2]:

return min(height[0],height[2]) - height[1]

return 0

for l in range(cnt):

if height[l] > 0: break

hl = height[l]

cur = l + 1

f = False

v = 0

while cur < cnt:

if height[cur] < hl:

f = True

else:

if f:

for i in range(cur-1, l, -1):

if height[i] < hl:

v += hl - height[i]

return self.trap(height[cur:]) + v

cur += 1

height.reverse()

return self.trap(height)