面试题解答
仅供学习
题目
要解决这个问题,可以使用双指针方法进行原地修改,以确保每个元素最多出现两次。
一、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,即新数组的长度。