【面试题解答】一个有序数组 nums ，原地删除重复出现的元素

题目

要解决这个问题，可以使用双指针方法进行原地修改，以确保每个元素最多出现两次。

一、python代码

1.1 代码

def removeDuplicates(nums):
    if not nums:
        return 0

    # 指针 j 用于维护新数组的位置
    j = 0
    # 当前元素计数器
    count = 1

    # 遍历数组
    for i in range(1, len(nums)):
        # 如果当前元素与前一个元素相同，则增加计数
        if nums[i] == nums[i - 1]:
            count += 1
        else:
            count = 1  # 重置计数器

        # 只保留当前元素的前两次出现
        if count <= 2:
            nums[j] = nums[i]
            j += 1

    return j

1.2 示例用法

1.2.1 示例1

nums1 = [1, 1, 1, 2, 2, 3]

length1 = removeDuplicates(nums1)

print(length1) # 输出: 5

print(nums1[:length1]) # 输出: [1, 1, 2, 2, 3]

1.2.2 示例2

nums2 = [0, 0, 1, 1, 1, 1, 2, 3, 3]

length2 = removeDuplicates(nums2)

print(length2) # 输出: 7

print(nums2[:length2]) # 输出: [0, 0, 1, 1, 2, 3, 3]

二、讲解

2.1 初始化

• j 指针用于跟踪新数组的位置。
• count 计数器用于跟踪当前元素出现的次数。

2.2 遍历

• 从第二个元素开始（i = 1），检查当前元素是否与前一个元素相同。
• 如果相同，则 count 增加；否则，重置 count 为 1。
• 如果 count 不超过 2，将当前元素放到 j 指向的位置，并移动 j 指针。

2.3 返回

• 函数返回 j，即新数组的长度。