题目描述
给定一个排序数组,你需要在原地删除重复出现的元素,使得每个元素只出现一次,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。
示例 1:
给定数组 nums = [1,1,2],
函数应该返回新的长度 2, 并且原数组 nums 的前两个元素被修改为 1, 2。
你不需要考虑数组中超出新长度后面的元素。
示例 2:
给定 nums = [0,0,1,1,1,2,2,3,3,4],
函数应该返回新的长度 5, 并且原数组 nums 的前五个元素被修改为 0, 1, 2, 3, 4。
你不需要考虑数组中超出新长度后面的元素。
此题要求在排好序的数组中,删除重复的元素;题目比较简单,但是有额外的要求,不能使用额外的数组空间。所以我们最直观的想法:“再新建一个数组B,遍历原数组A,每次判断A中遍历到的元素B中是否存在,不存在则添加”在这里不能使用。
解法:
这里使用双指针的思想来解决比较容易:创建慢指针和快指针,初始时两个指针同时指向数组中第一个元素,当两个指针所指数组元素相同时,快指针向后移动,慢指针不动;当快指针所指数组元素与慢指针不同时,慢指针向后移动,且将慢指针此时所指位置赋值为快指针所指元素值。当快指针指向数组中最后一个元素时结束。
在Python中没有指针的概念,不过可以使用for循环,flag标志位进行快、慢指针的模拟,具体代码如下:
class Solution(object):
def removeDuplicates(self, nums):
"""
:type nums: List[int]
:rtype: int
"""
# 考虑数组为空的情况
if len(nums) == 0:
return 0
flag = 0 # flag相当于慢指针
for num in nums: # for循环遍历数组元素,num相当于快指针
if nums[flag] != num:
flag += 1
nums[flag] = num
del nums[flag+1:]
return len(nums)
复杂度分析
-
时间复杂度:O(n),假设数组的长度是 n,那么 快指针和 慢指针分别最多遍历 n 步。
-
空间复杂度:O(1),因为在已有的数组空间进行操作(传入参数为数组),不引入额外的数组空间。