leetcode刷题删除排序数组中的重复项removeDuplicates(Python)

Posted 2022-02-08 IT界的测试混子

leetcode

删除排序数组中的重复项

给你一个有序数组 nums ，请你 原地 删除重复出现的元素，使每个元素 只出现一次 ，返回删除后数组的新长度。

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

说明:

   为什么返回数值是整数，但输出的答案是数组呢?
   请注意，输入数组是以「引用」方式传递的，这意味着在函数里修改输入数组对于调用者是可见的。
   你可以想象内部操作如下:
   // nums 是以“引用”方式传递的。也就是说，不对实参做任何拷贝 int len = removeDuplicates(nums);
   // 在函数里修改输入数组对于调用者是可见的。 // 根据你的函数返回的长度, 它会打印出数组中 该长度范围内 的所有元素。 for    (int i = 0; i < len; i++)     print(nums[i]); 

示例 1：

输入：nums = [1,1,2]
输出：2, nums = [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]
解释：函数应该返回新的长度 5 ， 并且原数组 nums 的前五个元素被修改为 0, 1, 2, 3, 4 。不需要考虑数组中超出新长度后面的元素。

提示：

0 <= nums.length <= 3 * 104shi de si ci fang
-104 <= nums[i] <= 104
nums 已按升序排列

作者：力扣 (LeetCode)
链接：https://leetcode-cn.com/leetbook/read/top-interview-questions-easy/x2gy9m/
来源：力扣（LeetCode） 著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

Solution1

for循环删除元素，其他元素的下标都会被改变。

• solution1 分两层循环，定义一个变量 存储上一个元素的值。
• 列表为空，则不进入内层循环。
• 内层循环一旦找到符合条件的元素，则删除。
• 删除则，跳出内层循环，继续从头开始循环新的列表。
• 如果循环到列表的最后一个元素，则flag为False，跳出所有循环。
• 程序结束。

class Solution:
    def removeDuplicates(self, nums) -> int:
        flag = True #循环标识
        while flag:
            if nums == []:
                break
            before = "" #存储前一个元素
            for i in nums:
                if nums.index(i) == len(nums)-1:
                    flag = False #如果循环到最后一个元素，flag为False结束循环
                if i == before:# 如果i等于上一个元素，则删除i，跳出本次循环，继续 从0循环列表
                    nums.remove(i)
                    break
                before = i #将 i 赋值给before
        return len(nums)

if __name__ == '__main__':
    nums = [[1,1,2],
            [0, 0, 1, 1, 1, 2, 2, 3, 3, 4],
            [],
            [1]
            ]
    exp = [(2,[1,2]),
           (5,[0,1,2,3,4]),
           (0,[]),
           (1,[1])
           ]
           
    s = Solution()
    for num in nums:
        print("------start------")
        print("num = %s " % str(num))
        res = s.removeDuplicates(num)
        print("exp = %s "%str(exp[nums.index(num)]))
        print("res = %s "%str((res,num)))
        print("result = %s "%str((res,num)==exp[nums.index(num)]))
        print("-------end-------")

Solution2

• 定义count存储删除元素次数
• 列表不为空则进入循环
• 循环列表index，如果上个元素值等于当前值且当前元素index非0（确保当前不是第一个元素），则删除当前元素，count加1

  def removeDuplicates2(self, nums) -> int:
        count = 0 #被删除的次数
        if nums:
            for i in range(len(nums)):
                if i>0 and nums[i-count] == nums[i-count-1]:
                    del nums[i-count]
                    count += 1
        return len(nums)

Solution3（最佳）

列表逆序删除，未被删除的元素下标不会被改变。避免了循环删除出现的下标改变的问题。

    def removeDuplicates3(self, nums) -> int:
        for i in range(len(nums) - 1, 0, -1):#逆序删除
            if nums[i] == nums[i - 1]:
                del nums[i]
        return len(nums)