26.删除排序数组中的重复项
题目
给定一个排序数组,你需要在 原地 删除重复出现的元素,使得每个元素只出现一次,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在 原地 修改输入数组 并在使用 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。
你不需要考虑数组中超出新长度后面的元素。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/remove-duplicates-from-sorted-array
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
题解思路
1>源代码
class Solution:
def removeDuplicates(self, nums: List[int]) -> int:
slow = 0
fast = 1
if (len(nums) == 1):
return 1
while (slow < len(nums)):
if (nums[slow] == nums[fast]):
fast += 1
if (fast > len(nums) - 1):
return slow + 1
else:
slow += 1
nums[slow] = nums[fast]
fast += 1
if (fast > len(nums) - 1):
return slow + 1
2>算法介绍
这道题的关键就在于,他需要你删除元素的数组是有序的!这里我们采用双指针的方式来解决问题,设置slow, fast两个指针。当slow和fast指向的值相同时,slow就原地等待,等待fast找到一个不相同的值,将该值放在slow+1指向的位置。举例说明:
[1, 2, 3, 3, 4]
当slow = 2, fast = 3的时候,slow指向的值和fast指向的值相等。
fast += 1,此时slow = 2, fast = 4。slow指向的值为3,fast指向的值为4,由于在上一个循环中已经发现slow指向的值存在重复,且第一次出现的位置为slow现在所指的位置,于是可以将slow值+1,此时slow指向的值是没有意义的,所以使用fast指向的值来替换这个值。
该例子最终变为:
[1, 2, 3, 4, 4]
由于题目告诉我们不需要考虑超出新长度后面的值,所以只需要返回此时slow + 1的值即可。