1. 程式人生 > >leetCode 11. 盛最多水的容器 (swift)

leetCode 11. 盛最多水的容器 (swift)

給定 n 個非負整數 a1,a2,...,an,每個數代表座標中的一個點 (iai) 。在座標內畫 n 條垂直線,垂直線 i 的兩個端點分別為 (iai) 和 (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

分析:從左右兩頭同時開始遍歷,哪邊的值小,哪邊就移動。

解:

class Solution {
    func maxArea(_ height: [Int]) -> Int {
        if height.count == 0 {
            return 0
        }
        var res = 0
        var left = 0
        var right = height.count - 1
        while left < right {
            let v = (right - left) * min(height[left], height[right])
            res = max(res,v)
            if height[left] < height[right]{
                left += 1
            }else{
                right -= 1
            }
        }
        return res
    }
}