leetcode 删除排序数组中的重复项 & 移除指定元素

悠悠 2022-03-27 12:54 152阅读 0赞

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

不要使用额外的数组空间，你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl8zNjM3Mjg3OQ_size_16_color_FFFFFF_t_70]

--------------------

算法思路:设置两个指针，i表示最后一个不重复的，i+1表示插入的位置，j一直往后面遍历。

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

![在这里插入图片描述][20190113191405165.png]

--------------------

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

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

元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl8zNjM3Mjg3OQ_size_16_color_FFFFFF_t_70 1]  
采用同样的思想

class Solution:
        def removeElement(self, nums, val):
            i = 0
            for j in range(0, len(nums)):
                if nums[j] != val:
                    nums[i] = nums[j]
                    i += 1
            return i

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl8zNjM3Mjg3OQ_size_16_color_FFFFFF_t_70]: /images/20220327/8d2aa4823631447b936845ab2c54387d.png
[20190113191405165.png]: /images/20220327/2e6e8191fcd74871ad6334a065940932.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl8zNjM3Mjg3OQ_size_16_color_FFFFFF_t_70 1]: /images/20220327/67b2b58dee7a41739e5857a0720c07f1.png