leetcode 11. 盛最多水的容器

 

 

 

New

 

给定 n 个非负整数 a1，a2，...，an，每个数代表坐标中的一个点 (i, ai) 。在坐标内画 n 条垂直线，垂直线 i 的两个端点分别为 (i, ai) 和 (i, 0)。找出其中的两条线，使得它们与 x 轴共同构成的容器可以容纳最多的水。

说明：你不能倾斜容器，且 n 的值至少为 2。

图中垂直线代表输入数组 [1,8,6,2,5,4,8,3,7]。在此情况下，容器能够容纳水（表示为蓝色部分）的最大值为 49。

 

示例:

输入: [1,8,6,2,5,4,8,3,7]
输出: 49

 

注意到如果从i开始，右边盛水最多的肯定是在j之后没有比j高的那个地方，所以可以做一个辅助数组，值为后一个最大的坐标的序列号。

典型的0，1背包，最大值为用了第一块木板和不用的最大值，递推式为D(x) = max(D(x+1) , maxD0)

func maxArea(height []int) int {
    if len(height) <= 1 {
        return 0
    }
    aux := make([]int, len(height))
    i := len(height) - 1
    maxnum := 0
    maxpos := 0
    for i >= 0 {
        if height[i] > maxnum {
            maxpos = i
            maxnum = height[i]
            aux[i] = -1
        } else {
            aux[i] = maxpos
        }
        i--
    }
    return maxSubArea(height, aux, 0)
}

func maxSubArea(height, aux []int, start int) int {
    // 0,1 backpack problem
    if len(height) - start < 2 {
        return 0
    }
    i := start
    max0 := 0
    for i < len(height) {
        if aux[i] > 0 {
            i = aux[i]
            continue
        } else {
            h := min(height[start], height[i]) * (i - start)
            if h > max0 {
                max0 = h
            }
            i++
        }
    }
    
    max1 := maxSubArea(height, aux, start+1)
    
    return max(max0, max1)
}

func max(a, b int) int{
    if a > b {
        return a
    } else {
        return b
    }
}

func min(a, b int) int {
    if a < b {
        return a
    } else {
        return b
    }
}