# 暴力法
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]))
