删除有序数组中的重复项Python3

题解:

class Solution:
    def removeDuplicates(self, nums: List[int]) -> int:
        # 如果列表为空,直接返回0
        if len(nums) == 0:
            return 0

        # 定义慢指针、快指针
        slow = fast = 0
        while fast < len(nums):
            # 如果慢指针指向的元素!=快指针指向的元素,
            # 说明慢指针可以向前一步,并赋值快指针指向的元素值 
            if nums[slow] != nums[fast]:
                slow += 1
                nums[slow] = nums[fast]  
            # 每循环一次,快指针都需要向前一步             
            fast += 1
        # 慢指针最终到的位置就是非重复元素最后的位置,元素个数=索引值+1
        return slow + 1

第一次在LeetCode刷题,看到如下的函数定义,搜索发现是python3.5新增的语法,用来说明参数和返回值的类型。

def removeDuplicates(self, nums: List[int]) -> int:

nums:List[int]表示参数nums的类型是List[int],->int表示返回类型是int。

详见文档:26.1. typing — Support for type hints — Python 3.5.9 documentation

由于题目要求在有序数字nums上原地操作,所以无法采用创建新的列表去遍历原列表的元素的方法,这里的解题方法用到的是双指针法

这个解题思路看了labuladong老师的算法小抄就感觉很好理解,这里放上链接:一文秒杀四道原地修改数组的算法题 - labuladong 的算法小抄

网上看到一张关于双指针法的图,感觉有助于理解,这里记录一下。

p、q分别表示慢指针、快指针,快指针走在前面。

如果q和p指向的元素一样,则q往前走;

直到找到和p不一样的,则p往前走,然后把p位置放q的值,然后q再往前走。

循环往复,直到q走到最后一个位置。

需要注意的是,如果列表为空,则直接返回0。


版权声明:本文为qq_29349463原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。