接雨水问题:双指针

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

 

 

接雨水问题:双指针_第1张图片

你可能感兴趣的:(接雨水问题:双指针)