Leetcode26. 删除有序数组中的重复数

一、题目描述：

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

由于在某些语言中不能改变数组的长度，所以必须将结果放在数组nums的第一部分。更规范地说，如果在删除重复项之后有 k 个元素，那么 nums 的前 k 个元素应该保存最终结果。

将最终结果插入 nums 的前 k 个位置后返回 k 。

不要使用额外的空间，你必须在原地修改输入数组 并在使用 O(1) 额外空间的条件下完成。

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

• 提示：
  • nums 已按 升序 排列
  • $-10^4\le nums[i]\le 10^4$
  • $1\le nums.length\le 3\ast 10^4$

二、解决思路和代码

1. 解决思路

• 分析：分两步进行，查找、替换。借助三个指针：start, right, end

  • 查找：right指针用来查找重复项，nums[start:right]为重复项
    

  • 替换：nums[right+1:end]替换掉nums[start+1:]

    

2. 代码

    from typing import *
    class Solution:
        def removeDuplicates(self, nums: List[int]) -> int:
            start = 0
            end   = len(nums)
            while start<end:
                right = start + 1
                while right<end and nums[right]==nums[start]: right += 1
                if right-start>1:
                    idx = start+1
                    while right<end:
                        nums[idx] = nums[right]
                        idx+=1
                        right+=1
                    end = idx
                start += 1
            nums = nums[:end]
            return len(nums)