【LeetCode.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。你不需要考虑数组中超出新长度后面的元素。

理解题意

• 原地修改数组，指的是直接对数组参数 取下标来修改数组元素，而不是通过将数组参数引用指向一个新的数组。
• 数组肯定是排序好的。
• 返回值为不重复元素个数。

思路

• 因为整个过程好像是在往左压实这个数组，所以我用了compact这个单词。图中我用蓝色标识当前的压实数组。
• 当新的压实元素加入压实数组时，是通过赋值来实现的。而赋值是会覆盖掉某个数组元素的，有两种情况：被覆盖的元素，本就是可以覆盖的，比如它是重复元素；或者，原地赋值，这样该索引上的元素不变。
• cursor代表遍历数组的索引，它从第二个元素开始，因为只有一个元素的数组肯定无重复呀。
• 在遍历过程中，cursor所在元素需要和compareNum进行比较，因为compareNum是当前已压实数组的最后一个元素，如果和它不一样，那说明cursor所在元素加入到压实数组中去：具体操作就是，把cursor所在元素赋值给压实数组之后的那个元素。
• 压实数组之后的那个元素的索引，即为compactIndex。

具体流程是：

• 遍历数组，如果cursor所在元素与compareNum相同，那么只增加cursor。
• 如果cursor所在元素与compareNum不同，那么除了增加cursor外，还需要压实操作：把cursor所在元素赋值给 compactIndex下标元素，更新compareNum为 cursor所在元素（因为现在它才是压实数组最后一个元素），当然compactIndex也得加1.
• 当然cursor到底末尾，结束循环。
• 当数组里没有重复元素时，你会发现有原地赋值的情况（即每次检测到新的压实元素时，cursor和compactIndex总是一样的），且每个新的压实元素，都是执行了原地赋值了的。

优化：
这是我最开始想的算法，其实聪明的你已经发现：compareNum和compactIndex总是相邻的，所以只需要保留一个就可以了。更进一步，compareNum你也不用用变量保存了，直接取下标就好了。

class Solution:
    def removeDuplicates(self, nums: List[int]) -> int:
        if len(nums) < 2:
            return len(nums)
        
        compactIndex = 1  #刚好是压实数组的长度
        compareNum = nums[0]
        cursor = 1
        while( cursor < len(nums) ):
            if nums[cursor] != compareNum:
                nums[compactIndex] = compareNum = nums[cursor]
                compactIndex += 1
            cursor +=1
        return compactIndex

解法关键词

• 双指针
• 快慢指针

最快代码

class Solution:
    def removeDuplicates(self, nums: List[int]) -> int:
        index = 0
        for i in range(1,len(nums)):
            if nums[i]  != nums[index]:  #每次和压实数组最后元素比较
                index +=1  #如果不同，index+1，因为新压实元素需要放在后面
                nums[index] = nums[i]  #赋值新压实元素
        return index+1 if nums else 0

• index为压实数组的最后一个元素的索引。
• 返回值为index+1，因为长度为最大索引+1.
• 没有保存compareNum了，但是有compareNum的索引index，每次取下标来用的。
• compactIndex也不用存了，因为它永远是index加1。
• 还有最后这个return index+1 if nums else 0，差点给我看蒙了。其实就是if else的倒装句和三目表达式的结合（nums ? index+1 : 0）。相当于下面：

if nums:
    index+1
else:
    0