题目
给你一个 升序排列 的数组 nums ,请你 原地 删除重复出现的元素,使每个元素 只出现一次 ,返回删除后数组的新长度。元素的 相对顺序 应该保持 一致 。
由于在某些语言中不能改变数组的长度,所以必须将结果放在数组nums的第一部分。更规范地说,如果在删除重复项之后有 k 个元素,那么 nums 的前 k 个元素应该保存最终结果。
将最终结果插入 nums 的前 k 个位置后返回 k 。
不要使用额外的空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
思路
采用快慢指针的做法。由于是有序数组,故只需要两两重复比较,随后将重复的数据删除即可。
实现代码
class Solution:
def removeDuplicates(self, nums: List[int]) -> int:
slow = 0
fast = 1
while fast< len(nums):
if(nums[slow] != nums[fast]):
slow += 1
fast += 1
else:
del nums[fast]
print(nums)
改进
这种情况下执行时间过久,参考评论区倒序遍历的思路,如果相同,直接pop出后一个值。
class Solution:
def removeDuplicates(self, nums: List[int]) -> int:
n = len(nums)
for i in range(n-1,0,-1):
if nums[i] == nums[i-1]:
nums.pop(i)
倒序而非正序的原因是,如果正序,数组变短的同时删除元素,会出现错误。