从排序数组中删除重复项

共计 1395 个字符，预计需要花费 4 分钟才能阅读完成。

题目描述

题目来自leetcode https://leetcode-cn.com/explore/interview/card/top-interview-questions-easy/1/array/21/

给定一个排序数组，你需要在原地删除重复出现的元素，使得每个元素只出现一次，返回移除后数组的新长度。

不要使用额外的数组空间，你必须在原地修改输入数组并在使用 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。

你不需要考虑数组中超出新长度后面的元素。

说明:

为什么返回数值是整数，但输出的答案是数组呢?

请注意，输入数组是以“引用”方式传递的，这意味着在函数里修改输入数组对于调用者是可见的。

你可以想象内部操作如下:

// nums 是以“引用”方式传递的。也就是说，不对实参做任何拷贝
int len = removeDuplicates(nums);

// 在函数里修改输入数组对于调用者是可见的。
// 根据你的函数返回的长度, 它会打印出数组中该长度范围内的所有元素。
for (int i = 0; i < len; i++) {
    print(nums[i]);
}

语言：python3

解法1: 单指针

比较相邻值，相等则删除前一个值，指针不后移，不等指针后移，循环比较

class Solution:
    def removeDuplicates(self, nums: List[int]) -> int:
        i = 0
        while i < len(nums)-1:
            if nums[i] == nums[i+1]:
              # 删除前一个值，保证nums[i]是重复值的第二个值，这样在遍历时顺序不会乱
                nums.remove(nums[i])  
            else:
              # 不相等则指针后移
                i += 1
        return len(nums)

解法2: 双指针（效率最高）

首先注意数组是有序的，那么重复的元素一定会相邻。

要求删除重复元素，实际上就是将不重复的元素移到数组的左侧。

考虑用 2 个指针，一个在前记作 j，一个在后记作 k，算法流程如下：

1.比较 j 和 k 位置的元素是否相等。

如果相等，j 后移 1 位
如果不相等，将 j 位置的元素复制到 k+1 位置上，j 后移一位，k 后移 1 位
这么做的目的是保证数组的第一个元素到k指针所在的元素为已排序且不重复的
重复上述过程，直到 j 等于数组长度。

返回 k + 1，即为新数组长度。

class Solution:
    def removeDuplicates(self, nums: List[int]) -> int:
        if not nums or len(nums) == 0:
            return 0
        j, k = 0, 0
        while j < len(nums):
            if nums[k] != nums[j]:
                nums[k+1] = nums[j]
                k += 1
            j += 1
        return k + 1

解法3: 利用python自建函数(效率最低)

class Solution:
    def removeDuplicates(self, nums: List[int]) -> int:
      # 去重，重新排序，转化为python list, 再赋值给数组
        nums[:] = list(sorted(set(nums))) 
        return len(nums)