接雨水问题:双指针

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

 

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值