题目
给定一个长度为 n 的整数数组 height 。有 n 条垂线,第 i 条线的两个端点是 (i, 0) 和 (i, height[i]) 。
找出其中的两条线,使得它们与 x 轴共同构成的容器可以容纳最多的水。
返回容器可以储存的最大水量。
说明:你不能倾斜容器。
示例 1:
输入:[1,8,6,2,5,4,8,3,7]
输出:49
示例 2:
输入:height = [1,1]
输出:1
答案
这道题可以使用双指针的方法解决,一个指针用来遍历数组,一个指针用来记录0的末尾位置。
具体步骤如下:
- 定义两个指针,一个指向当前遍历的位置,一个指向0的末尾位置。
- 遍历数组,如果当前元素为0,则将其移动到0的末尾位置,并将0的末尾位置指针向前移动一位。
- 如果当前元素不为0,则将其移动到当前位置,并将当前位置指针向前移动一位。
- 重复上述步骤,直到遍历完整个数组。
代码如下:
def moveZeroes(nums):if not nums:return []left = 0 # 当前遍历位置指针right = 0 # 0的末尾位置指针n = len(nums)while left < n:if nums[left] == 0:nums[left], nums[right] = nums[right], nums[left]right += 1else:nums[left] = nums[left] + 1 # 非零元素加上一个非零值,保证相对顺序不变left += 1return nums
时间复杂度为O(n),空间复杂度为O(1)。