[LeetCode小白]删除有序数组重复项Python

题目

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

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

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

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

思路

采用快慢指针的做法。由于是有序数组,故只需要两两重复比较,随后将重复的数据删除即可。

实现代码

class Solution:
    def removeDuplicates(self, nums: List[int]) -> int:
        slow = 0
        fast = 1
        while fast< len(nums):
            if(nums[slow] != nums[fast]):
                slow += 1
                fast += 1
            else:
                del nums[fast]
      
        print(nums) 

改进

这种情况下执行时间过久,参考评论区倒序遍历的思路,如果相同,直接pop出后一个值。

class Solution:
    def removeDuplicates(self, nums: List[int]) -> int:
       n = len(nums)
       for i in range(n-1,0,-1):
           if nums[i] == nums[i-1]:
               nums.pop(i)

倒序而非正序的原因是,如果正序,数组变短的同时删除元素,会出现错误。


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