Leetcode --- 删除排序数组中的重复项 和 移除元素（python3实现）

题目描述

1.删除排序数组中的重复项

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

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

示例 1:

给定数组 nums = [1,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。

你不需要考虑数组中超出新长度后面的元素。

 

2.移除元素

给定一个数组 nums 和一个值 val，你需要原地移除所有数值等于 val 的元素，返回移除后数组的新长度。

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

元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。

示例 1:

给定 nums = [3,2,2,3], val = 3,

函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。

你不需要考虑数组中超出新长度后面的元素。
示例 2:

给定 nums = [0,1,2,2,3,0,4,2], val = 2,

函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。

注意这五个元素可为任意顺序。

你不需要考虑数组中超出新长度后面的元素。

题目分析

根据题目要求，可总结出以下几点：

1.需要原地修改数组

2.返回处理后的个数(n)，且保证nums[:n] 为处理后的元素列表

3.不使用额外的数组空间

结合以上几点，可以采用指针的概念来实现这两道题，通过指针指向来原地替换掉nums[:n]的元素

题解

1.删除排序数组中的重复项

class Solution:
    '''
    删除排序数组中的重复项
    思路---> 1.根据题目提示判断-->1.需要原地修改数组;
                              2.返回不重复项个数;
                              3.不使用额外的数组空间，即不允许重新创建一个数组，然后赋值给新数组;
            2.如：输入为nums = [0,1,1,1,3,3,4] , 返回不重复个数为 n = 4 , 且保证 nums[:n] 这段元素为不重复数
    方法---> 1.不重复元素的个数一定小于原列表长度，所以为了保证不使用额外的数组空间和原地修改数组，只需要替换掉前 n 个元素就行了 (n 为不重复的个数)
    '''
    def removeDuplicates(self, nums):
        nums_len = 1
        for x, y in enumerate(nums):
            if x <= len(nums) - 2:                 # 避免超出列表长度
                if y != nums[x + 1]:               # 将当前元素与下一个元素相比较
                    nums[nums_len] = nums[x + 1]   # 替换掉nums_len指向的原数组元素
                    nums_len += 1                  # 推进 ---> 没遇到不同的元素前，不推进。用于逐步修改原数组的值，实现 nums[:nums_len] 为不重复元素
            # print(nums)
        return nums_len

li = [0,1,1,1,3,3,4]
result = Solution().removeDuplicates(li)
print(result)

2.移除元素

class Solution:
    '''
    移除元素
    思路---> 1.根据题目提示判断-->1.需要原地修改数组;
                              2.返回移除后的元素个数；
                             3.不使用额外的数组空间，即不允许重新创建一个数组，然后赋值给新数组;
            2.如：输入为nums = [3，2，2，3] , 返回剩余个数 n = 2 , 且保证 nums[:n] 这段元素为剩余元素
    方法---> 1.移除之后的元素个数一定小于原列表长度，所以为了保证不使用额外的数组空间和原地修改数组，只需要替换掉前 n 个元素就行了 (n 为没移除元素个数)
    '''
    def removeElement(self, nums,val):
        init_num = 0                       # 初始指针指向为数组的首位
        for i in range(len(nums)):
            if nums[i] != val:             # 将数组元素与给定值val比较
                nums[init_num] = nums[i]   # 如果该数组元素与给定值val不相同，将指针指向的元素替换为该数组元素
                init_num += 1              # 推进指针指向
            # print(nums)
        return init_num

li = [0,1,2,2,3,0,4,2]
num = 2
result = Solution().removeElement(li,num)
print(result)