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

思路

定义一个变量count，记录不相同的元素的个数，遍历数组，如果前后两个元素相同，则继续往下走，如果不同，则count+1，并且在索引为count的位置上直接修改为第i个元素的值，因为count是从0开始的，最后返回长度时需要加1

class Solution(object):
    def removeDuplicates(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        if len(nums)<1:return 0
        count=0
        for i in range(1,len(nums)):
            if nums[i] != nums[i-1]:count+=1;nums[count]=nums[i]
        return count+1