题目:
给你一个有序数组 nums ,请你 原地 删除重复出现的元素,使每个元素只出现一次 ,返回删除后数组的新长度。
不要使用额外的数组空间,你必须在原地修改输入数组,并在使用 O(1) 额外空间的条件下完成。
题目地址:删除排序数组中的重复项
题目条件限制:
该数组为有序数组(已排序)
在数组中删除重复出现的元素,使每个元素只出现一次,既:nums.sort()
返回删除后的数组长度,既:len(nums)
空间复杂度为O(1),即只用常数的额外空间
时间复杂度为O(n),既双指针各只能遍历一次
解题思路:
- 数组已排序,重复元素的只出现在目标元素的两侧且连续
- 数组数量为0时,直接返回0
- 数组数量为1时,无重复元素
- 数组从nums[2]开始才会有重复元素出现
- 利用双指针解法可实现只遍历一次的情况下记录非重复元素,慢指针作游标和记录功能,快指针作遍历功能
- 设 nums 长度为 n ,既 len(nums),快指针为 j ,慢指针为 i
- 快、慢指针从第0个元素开始(nums[i]=nums[j]=nums[0])
- 快指针先向前移动一位,既nums[j] = nums[1],此时,慢指针nums[i] = 0
- 相互比较快慢指针的元素,nums[i] = nums[j] 既 (nums[0] = nums[1])
- 若快指针移动一位后仍与慢指针元素相同,既证明此时的 nums[i] 和 nums[j] 为相同元素,反之,如果不相等,则记录此时的慢指针所指的元素 nums[i]并将慢指针向前移动一位
- 条件判断后(既判断慢指针是否记录和移动),都将快指针继续向前移动一位
- 以快指针先移动,双指针作比较(满足条件则触发条件),为周期循环,直至双指针持续遍历n-1个元素,如下图
官方代码
class Solution:
def removeDuplicates(self, nums: List[int]) -> int:
if not nums: # 特殊情况,当nums为空集时
return 0
n = len(nums)
fast = slow = 1 # nums的第一个元素必定会保留,所以从第二个元素开始判断是否与前面的元素相同
while fast < n: # 既fast = n-1 时就无法循环,从n[0]遍历到n[n-1]刚好是n次;n = len(nums)
if nums[fast] != nums[fast - 1]: # 利用了重复元素必定是连续的,这个条件
nums[slow] = nums[fast] # 当fast不等于前一个元素时,nums[slow]当前的位置就等于不重复元素
slow += 1 # 慢指针移动到下一位,既下一个nums[slow]
fast += 1 # 快指针移动到下一位,既下一个nums[fast]
return slow # 返回了nums[0]到nums[slow-1]个元素
官方测试的数组为:nums = [1,1,2]
自己带入数组,尝试理解一下
再带入一个数组nums = [1,2,3,3],就好理解了
网友的优秀代码
# 通解扩展
class Solution:
def removeDuplicates(self, nums: List[int]) -> int:
def solve(k): # 最多保留k位相同数字
slow = 0 # 慢指针从0开始
for fast in nums: # 快指针遍历整个数组
# 检查被保留的元素nums[slow−k]是否和当前待检查元素fast相同
if slow < k or nums[slow - k] != fast:
nums[slow] = fast
slow += 1
return slow # 从nums[0]到nums[slow−1]的每个元素都不相同
return solve(2)
网友把题目转化为“每个元素最多出现K次”,再把K = 2,就能解答了
结论:
本题主要讲解双指针的基本用法,题目限制了很多的条件,个人理解为为了代码而设计的题目