python每日一题——5盛最多的水

题目

给定一个长度为 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的末尾位置。

具体步骤如下:

  1. 定义两个指针,一个指向当前遍历的位置,一个指向0的末尾位置。
  2. 遍历数组,如果当前元素为0,则将其移动到0的末尾位置,并将0的末尾位置指针向前移动一位。
  3. 如果当前元素不为0,则将其移动到当前位置,并将当前位置指针向前移动一位。
  4. 重复上述步骤,直到遍历完整个数组。

代码如下:

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 += 1
        else:
            nums[left] = nums[left] + 1  # 非零元素加上一个非零值,保证相对顺序不变
            left += 1
    return nums

时间复杂度为O(n),空间复杂度为O(1)。

你可能感兴趣的:(算法练习,python,算法,leetcode)