删除排序数组中的重复项

题目描述:

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

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

示例 :

给定数组 nums = [1,1,2], 
 
函数应该返回新的长度 2, 并且原数组 nums 被修改为[1,2]
给定数组 nums = [0,0,1,1,1,2,2,3,3,4],
 
函数应该返回新的长度 5, 并且原数组 nums 被修改为 [0,1,2,3,4]

题目分析:

看到这题目,不看题目的具体要求,第一想到的是用 set 函数,set 和 dict 类似,也是一组 key 的集合,但不存储 value。由于 key 不能重复,所以,在 set 中,没有重复的 key。利用这一特性,可以过滤掉重复元素。

但是仔细一看题目要求,完全不符合,题目是要求原地删除重复元素。如果用 set 方法,会返回一个新的字典,而原数组没有改变,结果如下:

>>> nums = [1,1, 2, 3, 3, 3, 5, 6] 
>>> s = set(nums)

>>> nums

[1, 1, 2, 3, 3, 3, 5, 6]

>>> s

{1, 2, 3, 5, 6}

跑偏了,回到题目本身。

list 有个 remove 方法,remove 方法是一个没有返回值的原位置改变的方法,他修改了列表但是没有返回值。

由于给定的是排序数组,这就容易多了,比较两两相邻的数组元素,相同就把其中的一个利用 remove 方法从列表中移除。

class Solution(object):
    def removeDuplicates(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        list_index = 0
        
        while list_index < len(nums) - 1:
            if nums[list_index] == nums[list_index + 1]:
                nums.remove(nums[list_index])
            else:
                list_index += 1
                
        return len(nums)