题干:
给你一个有序数组 nums
,请你 原地 删除重复出现的元素,使得出现次数超过两次的元素只出现两次 ,返回删除后数组的新长度。
不要使用额外的数组空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
解法1:快慢指针
func removeDuplicates(nums []int) int {
if len(nums) <= 2 {
return len(nums)
}
i := 1 // 初始化 i 为 1,因为前两个元素无论如何都会保留
for j := 2; j < len(nums); j++ {
if nums[j] != nums[i] || nums[j] != nums[i-1] {
i++
nums[i] = nums[j]
}
}
return i + 1
}
解析1:
这道题是26.删除有序数组中的重复项的续集。
原地删除就是暗示替换元素的意思。
两道题的相同点:有序数组意味着重复项只能相邻
i
是慢指针,表示当前有效数组的最后一个元素的索引。初始化为1,表示当前有效数组的最后一个元素的索引。前两个元素无论如何都会保留。j
是快指针,从索引2开始遍历数组,直到数组末尾。- 对于每个元素
nums[j]
,检查它是否与nums[i]
和nums[i-1]
不同。 - 如果
nums[j]
不等于nums[i]
或nums[j]
不等于nums[i-1]
,则将nums[j]
放到nums[i+1]
的位置,并更新i的位置
。这样可以确保每个元素最多出现两次。
- 对于每个元素
复杂度分析:
- 时间复杂度:O(n),其中 n 是数组的长度。我们只需要遍历数组一次。
- 空间