1. 程式人生 > 實用技巧 >leetcode之11盛3最多水的容器Golang

leetcode之11盛3最多水的容器Golang

問題描述:

給你 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。

解題思路:

本題採用雙指標的方法,首先將指標設定在首尾,這樣就保證了矩形的長度最長了,然後接下來再將首尾指標向中間移動,直到他們相遇

在長度一定的的情況下,矩形的面積由高度決定,高度由較短的高度決定,所以每次我們向中間移動較短的高,並且更新最大面積

程式碼如下:

func maxArea(height []int) int {
	min := func(a, b int) int {
		if a <= b {
			return a
		}
		return b
	}
	h, t := 0, len(height)-1
	maxWater := t * min(height[h], height[t])
	for h != t {
		if height[h] <= height[t] {
			h++
		} else {
			t--
		}
		tmpWater := (t - h) * min(height[h], height[t])
		if tmpWater > maxWater {
			maxWater = tmpWater
		}
	}
	return maxWater
}