击退 LeetCode，算法巧劲化解重复项 II，拥抱有序数组的真谛

后端

2024-01-08 23:06:57

从有序数组中删除重复项：LeetCode 80 攻关指南

欢迎来到算法世界的又一轮激动人心的旅程！今天，我们深入探究 LeetCode 80——删除有序数组中的重复项 II。在这个任务中，我们的目标是消除给定有序数组中的所有重复项，但最多保留两次。让我们一起踏上算法之旅，揭开谜题的奥秘。

双指针法：破译谜题的关键

要成功解开 LeetCode 80，我们求助于算法领域的得力助手——双指针法。想象有两个指针，一个名为 "slow"，另一个名为 "fast"，它们沿着数组移动，执行以下任务：

慢速指针（slow）： 负责标记新数组中的元素位置。
快速指针（fast）： 负责探索数组，寻找与 slow 指针指向元素不同的元素。

算法步骤：

初始化指针： 将 slow 和 fast 指向数组的第一个元素。
移动快速指针： 向右移动 fast 指针，直到找到一个与 slow 指针指向元素不同的元素。
更新慢速指针： 当找到不同的元素时，将 slow 指针移到该元素上。
重复步骤 2 和 3： 继续移动 fast 指针并更新 slow 指针，直到 fast 指针到达数组末尾。
返回结果： slow 指针现在指向新数组的末尾，其中只包含最多保留两次的重复项。

代码实现：用 Python 征服 LeetCode 80

掌握了算法的精髓，让我们用 Python 赋予其生命：

def remove_duplicates(nums):
  """
  Removes duplicate elements from a sorted array, allowing up to two duplicates.

  Parameters:
    nums: A sorted array of integers.

  Returns:
    The length of the modified array.
  """
  slow = 0
  fast = 0
  count = 0

  while fast < len(nums):
    if nums[fast] == nums[slow]:
      count += 1
      if count <= 2:
        slow += 1
        nums[slow] = nums[fast]
    else:
      count = 1
      slow += 1
      nums[slow] = nums[fast]

    fast += 1

  return slow + 1