题目要求:
给定一个长度为 n 的整数数组 height 。有 n 条垂线,第 i 条线的两个端点是 (i, 0) 和 (i, height[i]) 。
找出其中的两条线,使得它们与 x 轴共同构成的容器可以容纳最多的水。
返回容器可以储存的最大水量。
说明:你不能倾斜容器。
示例 1:
输入:[1,8,6,2,5,4,8,3,7]
输出:49
解释:图中垂直线代表输入数组 [1,8,6,2,5,4,8,3,7]。在此情况下,容器能够容纳水(表示为蓝色部分)的最大值为 49。
示例 2:
输入:height = [1,1]
输出:1
解题思路
主要使用双指针,一个从坐标0开始往右遍历,一个从右边(height)开始,计算两个指针的乘积,哪个最大,返回哪个,需要注意的是go没有内置min和max等内置比较函数,需要自己实现,也可以调用math.Min和math.Max
代码如下:
func maxArea(height []int) int {
// 使用双指针
ans := 0
left, right := 0, len(height)-1
for left < right {
area := (right - left) * minFunc(height[left], height[right])
ans = maxFunc(ans, area)
if height[left] < height[right] {
left++
} else {
right--
}
}
return ans
}
func minFunc(a, b int) int {
if a > b {
return b
}
return a
}
func maxFunc(a, b int) int {
if a < b {
return b
}
return a
}