接雨水问题:双指针
# 暴力法 def rain(lis): n = len(lis) l_max = [0 for _ in range(n)] # 记录每个点为中心,左边和右边最高的板高度 r_max = [0 for _ in range(n)] for i in range(1, n): l_max[i] = max(l_max[i - 1], lis[i - 1]) for j in range(n - 2, -1, -1): r_max[j] = max(r_max[j + 1], lis[j + 1]) res = 0 for k in range(n): tmp = min(l_max[k], r_max[k]) - lis[k] # 记录每个点的储水量,有可能出现负值,除去 res += tmp if tmp > 0 else 0 return res print(rain([0,1,0,2,1,0,1,3,2,1,2,1])) # 双指针 def point_search(lis): n = len(lis) left, right = 0, n - 1 l_max, r_max = lis[0], lis[-1] res = 0 while left <= right: # 通过最左边和最右边的指针不断向中间移动,计算每个点上面的储水量(大于等于0) l_max = max(l_max, lis[left]) r_max = max(r_max, lis[right]) if l_max < r_max: # 取两边较低的高度计算 res += l_max - lis[left] left += 1 else: res += r_max - lis[right] right -= 1 return res print(point_search([0,1,0,2,1,0,1,3,2,1,2,1]))