1:题目描述(力扣)
给你一个 升序排列 的数组 nums ,请你 原地 删除重复出现的元素,使每个元素 只出现一次 ,返回删除后数组的新长度。元素的 相对顺序 应该保持 一致 。
由于在某些语言中不能改变数组的长度,所以必须将结果放在数组nums的第一部分。更规范地说,如果在删除重复项之后有 k 个元素,那么 nums 的前 k 个元素应该保存最终结果。
将最终结果插入 nums 的前 k 个位置后返回 k 。
不要使用额外的空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
2:解题思路
使用双指针解法
第一步:定义两个指针,slow(慢指针),fast(快指针)
第二步:遍历数组,判断fast指针对应数组元素与slow指针对应的数据元素是否相等,
nums[fast] != nums[slow],需要将slow指针向右移动,并将fast指针对应的元素赋值给此时下标为slow位置
遍历下一个元素,fast指针向右移动
当nums[fast] = nums[slow],遍历下一个元素,如下图
nums[fast] != nums[slow],slow+=1,nums[slow]=nums[fast]
第三步:遍历完数组,返回slow+1
为什么是slow+1,而不是slow呢?
因为数组的下标是从0开始的,遍历完数组后slow的值为5,新数组的长度为6,所以返回slow+1
代码展示:
class Solution:
def removeDuplicates(self, nums):
slow = 0
for fast in range(1, len(nums)):
if nums[fast] != nums[slow]:
slow += 1
nums[slow] = nums[fast]
return slow+1
上述代码可以进行优化:
考虑如下数组:
此时数组中没有重复元素,按照上面的方法,每次比较时 nums[slow] 都不等于 nums[fast],因此就会将 fast指向的元素原地复制一遍,这个操作其实是不必要的。
因此我们可以添加一个小判断,当 fast - slow > 1 时,才进行复制。
优化代码如下:
class Solution:
def removeDuplicates(self, nums):
slow = 0
for fast in range(1, len(nums)):
if nums[fast] != nums[slow]:
if fast - slow > 1:
nums[slow+1] = nums[fast]
slow += 1
return slow+1