go语言LeetCode刷题笔记
Hot100系列
第11题盛最多水的容器
给定一个长度为 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
提示:
n == height.length
2 <= n <= 105
0 <= height[i] <= 104
思路:
我们可以知道,蓄水面积的公式等于
A
r
e
a
=
m
i
n
(
l
e
f
t
,
r
i
g
h
t
)
×
(
r
i
g
h
t
)
Area=min(left, right) \times (right)
Area=min(left,right)×(right)
right表示右边挡板的位置,left表示左边挡板的位置。蓄水容量应为左右挡板的最小高度(短板效应hhhh)乘以他们之间的距离,也就是right-left。
那么接下来应该怎么算呢?
当然是移动最小高度的挡板,继续寻找是否有能存储更多水的方案。
为什么是移动最小高度的挡板呢?因为从公式中我们可以看出,蓄水面积只和两个挡板中最小高度的挡板有关。
那么至此,我们便可以得到写出代码。
go语言实现代码如下
func maxArea(height []int) int {
res := 0
left, right := 0, len(height)-1
for left < right {
if height[left] < height[right] {
res = max(res, (right - left) * height[left])
left++
}else{
res = max(res, (right - left) * height[right])
right--
}
}
return res
}
func max(a, b int) int {if a > b {return a}; return b}
我们使用left和right当做左右挡板,用res记录最大蓄水量,于是我们只需要遍历一遍数组便可得到结果。时间复杂度是O(n),而且也没有使用额外空间,空间复杂度是O(1)。