【LeetCode刷题】26. 删除有序数组中的重复项

1:题目描述(力扣)

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

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

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

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

【LeetCode刷题】26. 删除有序数组中的重复项_第1张图片

2:解题思路

使用双指针解法

第一步:定义两个指针,slow(慢指针),fast(快指针)

【LeetCode刷题】26. 删除有序数组中的重复项_第2张图片

 第二步:遍历数组,判断fast指针对应数组元素与slow指针对应的数据元素是否相等,

                nums[fast] != nums[slow],需要将slow指针向右移动,并将fast指针对应的元素赋值给此时下标为slow位置

【LeetCode刷题】26. 删除有序数组中的重复项_第3张图片

        遍历下一个元素,fast指针向右移动

【LeetCode刷题】26. 删除有序数组中的重复项_第4张图片        当nums[fast] = nums[slow],遍历下一个元素,如下图

【LeetCode刷题】26. 删除有序数组中的重复项_第5张图片

        nums[fast] != nums[slow],slow+=1,nums[slow]=nums[fast]

【LeetCode刷题】26. 删除有序数组中的重复项_第6张图片

 【LeetCode刷题】26. 删除有序数组中的重复项_第7张图片

 【LeetCode刷题】26. 删除有序数组中的重复项_第8张图片

【LeetCode刷题】26. 删除有序数组中的重复项_第9张图片 【LeetCode刷题】26. 删除有序数组中的重复项_第10张图片

第三步:遍历完数组,返回slow+1

为什么是slow+1,而不是slow呢?

因为数组的下标是从0开始的,遍历完数组后slow的值为5,新数组的长度为6,所以返回slow+1

代码展示:

class Solution:
    def removeDuplicates(self, nums):
        slow = 0
        for fast in range(1, len(nums)):
            if nums[fast] != nums[slow]:
                slow += 1
                nums[slow] = nums[fast]
        return slow+1

 上述代码可以进行优化:

考虑如下数组:

【LeetCode刷题】26. 删除有序数组中的重复项_第11张图片

此时数组中没有重复元素,按照上面的方法,每次比较时 nums[slow] 都不等于 nums[fast],因此就会将 fast指向的元素原地复制一遍,这个操作其实是不必要的。

因此我们可以添加一个小判断,当 fast - slow > 1 时,才进行复制。

优化代码如下:

class Solution:
    def removeDuplicates(self, nums):
        slow = 0
        for fast in range(1, len(nums)):
            if nums[fast] != nums[slow]:
                if fast - slow > 1:
                    nums[slow+1] = nums[fast]
                slow += 1
        return slow+1

你可能感兴趣的:(Leetcode,leetcode,算法,python)