python-蓄水池问题

614 阅读1分钟

蓄水池

题目:给定一个列表,列表中的元素表示蓄水池的木板的高度。如果以这些木板组合为一个水池(宽度为1),最多可以容纳多少单位的水。

题解:

遍历每一块木板的高度,然后查看当前木板的左边和右边的各自最高的木板的高度,然后取两者的较小值,然后跟当前木板的高度作比较,如果比当前木板高,则表示当前木板这里可以积累(min( max_{left}, max_{right} ) - cur)体积的水。

#coding:utf-8
 
# 题目介绍:给定一个数组,该数组表示每一个单位长度的石柱的高度,请问当前
# 的石柱围城的空间内,可以容纳多少的水
 
def finmax(subarray):
    """
    查找子序列中的最大值
    """
    print(subarray)
    return max(subarray)
 
def findLeftMax(array, index):
    """
    查找子序列arry[0:index]的最大值
    """
    subarray = array[0:index]
    return finmax(subarray)
 
def findRightMax(array, index):
    """
    查找子序列array[index + 1:]的最大值
    """
    subarray = array[index + 1:]
    return finmax(subarray)    
 
def run(array):
    """
    """
    length = len(array)
 
    sum = 0
    for i in range(length):
        if i == 0 or i == (length - 1):
            continue
        leftmax = findLeftMax(array, i)
        rightmax = findRightMax(array, i)
 
        maxvalue = min(leftmax, rightmax)
        if maxvalue - array[i] > 0:
            sum = sum + (maxvalue - array[i])
    print("the maxcontainer is %s" % sum)
 
if __name__ == "__main__":
    array = [1,8,6,2,5,4,8,3,7]
    run(array)