首页手记 LeetCode 26. 删除排序数组中的重复项

LeetCode 26. 删除排序数组中的重复项

标签：

Python 算法

26. 删除排序数组中的重复项

题目来源：https://leetcode-cn.com/problems/remove-duplicates-from-sorted-array

题目

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

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

解题思路

思路：双指针

这里要注意审题，题目中已经说明，数组是已经排过序的。那么根据这个前提，我们就能够得出一点，如果数组中，有重复项，那么重复项一定是相邻的。

根据这个前提，我们初始化两个指针，一个在前 front，一个在后 tail。front 先移动，如果 nums[front] == nums[tail]，表示元素值重复，front 直接向后移动。

如果 nums[front] != nums[tail]，那么表示两个元素不再是重复项，此时将 front 指针所对应的元素值复制给 nums[tail] 的下一个元素，也就是 nums[tail+1]，同时 tail 也要向前移动。

重复上面的过程，直到 front 到达数组尾部。

具体的过程可看图解：

代码实现

class Solution:
    def removeDuplicates(self, nums: List[int]) -> int:
        length = len(nums)
        if length == 0:
            return 0
        
        # 初始化双指针
        # front 在前，tail 在后
        tail = 0
        front = 1
        
        # 遍历数组
        while front < length:
            # 如果两指针所指的元素值不相同，
            # 表示元素不重复，
            # 这时候将 nums[front] 的值复制给 nums[tail+1]
            if nums[front] !=  nums[tail]:
                tail += 1
                nums[tail] = nums[front]
            # front 持续移动
            front += 1

        return tail + 1