题目描述:
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/remove-duplicates-from-sorted-arrayhttps://leetcode-cn.com/problems/remove-duplicates-from-sorted-array
给你一个 升序排列 的数组 nums ,请你 原地 删除重复出现的元素,使每个元素 只出现一次 ,返回删除后数组的新长度。元素的 相对顺序 应该保持 一致 。
由于在某些语言中不能改变数组的长度,所以必须将结果放在数组nums的第一部分。更规范地说,如果在删除重复项之后有 k 个元素,那么 nums 的前 k 个元素应该保存最终结果。
将最终结果插入 nums 的前 k 个位置后返回 k 。
不要使用额外的空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
示例 1:
输入:nums = [1,1,2]
输出:2, nums = [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]
解释:函数应该返回新的长度 5 , 并且原数组 nums 的前五个元素被修改为 0, 1, 2, 3, 4 。不需要考虑数组中超出新长度后面的元素。
一开始就把问题想的复杂了,思路也是双指针
初始思路:
1.i指向第一位,j从i之后遍历,j指向的若与i相同则删除(修改索引位置为某个标志位,如None,null),指导下一个不等于i指向元素时,i指向j,j再从此时的i之后开始向后遍历直到i到末尾之前一位
2.完成重复数组删除之后,此时数组是有标志位碎片的,要将其紧凑至前n位。从头至尾找到第一个空位,再找下一个非空,则空位赋值为第一个找到的非空元素,并将非空元素删除,向后寻找直至空位之后直到末尾没有非空,则循环结束
代码如下:
def removeDuplicates(self, nums: List[int]) -> int:
# 两个指针从尾至头遍历数组 找到重复则删除
# 完成删除之后需要紧凑数组,从头至末尾,找到第一个为空的位置,
# 再找下一个非空位置,赋值并删除,随后寻找下一个非空的位置,
# 若找到空位之后,其后全部为空,则停止循环
i = 0
j = 1
while i < len(nums) - 1:
while j < len(nums):
if nums[i] == nums[j]:
# 相等则删除后者 随后继续向后寻找
nums[j] = None
j += 1
else:
# 不相等则i从j的当前位置开始寻找 跳出当前循环
i = j-1
break
i += 1
j = i+1
# 缩紧元素
for i in range(0, len(nums)):
if nums[i] is not None:
continue
else:
for j in range(i + 1, len(nums)):
if nums[j] is not None:
nums[i] = nums[j]
nums[j] = None
break
else:
continue
length = 0
for i in range(0, len(nums)):
if nums[i] is not None:
length += 1
return length
此段代码时间复杂度达到了o(n²),在数组特别大的时候运算时间超时,显然有些笨。
随后想到造轮子不如用轮子,集合(set)具有天然的去重属性,在list和set之间倒腾一下就完事了其实,这样反而简单了,代码如下:
# 集合
nums_set = set()
for x in nums:
nums_set.add(x)
length_of_set = len(nums_set)
nums.clear()
nums.extend(nums_set)
nums.sort()
return length_of_set
但是这样就没有了算法题的意义了,真正正确的解法是使用快慢指针
数组有序,两个相等的值之间的所有值均相等,若数组长度为零,则返回0;大于零时数组中至少有一个元素,因此从下表1开始即可。
定义两个指针 fast和 slow 分别为快指针和慢指针,快指针表示遍历数组到达的下标位置,慢指针表示下一个不同元素要填入的下标位置,初始时两个指针都指向下标 1
假设数组长度为n,快指针遍历1到n-1,每个位置与其前一位相比,如果nums[fast] != nums[fast-1]
说明nums[fast]和之前的元素都不相同,则将其赋值给慢指针slow所指向的元素。然后将慢指针向后移一位,快指针此时仍从当前位置向后遍历,所以时间复杂度是O(n),直到末尾,则慢指针最后之前的元素就是去重之后的元素,返回slow即为数组长度,代码如下:
class Solution:
def removeDuplicates(self, nums: List[int]) -> int:
if not nums:
return 0
n = len(nums)
fast = slow = 1
while fast < n:
if nums[fast] != nums[fast - 1]:
nums[slow] = nums[fast]
slow += 1
fast += 1
return slow