[ LeetCode ] 问题刷（Python） - 问题 26：从有序数组中删除重复数据

题目描述

给你一个 非严格递增排列 的数组 nums ，请你 原地 删除重复出现的元素，使每个元素 只出现一次 ，返回删除后数组的新长度。元素的 相对顺序 应该保持 一致 。然后返回 nums 中唯一元素的个数。

考虑 nums 的唯一元素的数量为 k ，你需要做以下事情确保你的题解可以被通过：

• 更改数组 nums ，使 nums 的前 k 个元素包含唯一元素，并按照它们最初在 nums 中出现的顺序排列。nums 的其余元素与 nums 的大小不重要。
• 返回 k 。

示例

示例 1：

输入：nums = [1,1,2]
输出：2, nums = [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]
解释：函数应该返回新的长度 5 ， 并且原数组 nums 的前五个元素被修改为 0, 1, 2, 3, 4 。不需要考虑数组中超出新长度后面的元素。

解题

解法一： 双指针法

思路

要求原地修改数组，那么我们从第二个元素开始遍历进行处理，就先假设第一个元素为不重复元素啦，人家是第一耶。所以它就是新数组的第一个元素。

在遍历过程中，判断当前的原数组元素是否和新数组的最新元素相等，若不同，则将其填入新数组的下一个位置（索引为j+1）并将j递增1。

nums这个数组是有序的，所以重复的元素是相邻的。后面处理的原数组元素不可能会和新数组的非最新元素重复，只可能和新数组的最新元素重复。

算法复杂度

时间复杂度：O(n)，其中n为nums的长度。

---

空间复杂度：O(1)。

代码

class Solution:
    def removeDuplicates(self, nums: List[int]) -> int:
        if not nums:
            return 0
        j = 0  # 记录新数组的下标，原来数组的第一个元素不用处理
        for i in range(1, len(nums)):
            # 如果原数组当前元素和新数组最新元素不同
            if nums[i] != nums[j]:
                j += 1   # 新数组的下一个位置下标
                nums[j] = nums[i] # 将原数组中新的不同元素放入新数组的下一个位置

        return j + 1  # 返回新数组的长度（下标从0开始，所以长度=最后的下标+1）