Leetcode_27 Remove Element

发表于 | 分类于

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

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

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

示例 1:

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

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

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

1
2
3
4
5
6
7
8
9
示例 2:

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

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

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

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

详解：

分析：给定一个数组，去除数据中给定的数，空间复杂度限制在O(1)，不要求顺序一致

算法：
第一种方式：
通过两个指针i,j
i:新的数组下标，起始为0
j:遍历原数组
当出现不需要去除的数(nums[j]!=val),指针nums[i]指向该数，同时i增加1
最后i为新的数组的长度

1
2
3
4
5
6
7
8
9
10
11
12
13
class Solution:
    def removeElement(self, nums, val):
        """
        :type nums: List[int]
        :type val: int
        :rtype: int
        """
        i = 0
        for j in range(len(nums)):
            if nums[j] != val:
                nums[i] = nums[j]
                i += 1
        return i

复杂度分析：

时间复杂度：O(n):假设数组总共有n个元素，i和j最多遍历2n次操作。

空间复杂度：O(1)

第二种方式：

第一种方式对于数据中去除的数很少的情况下不够优雅。例如[1,2,3,4,5]中去掉5，多了1-4无意义的复制动作。

算法（顺序不要求一致）：
每当找到nums[i] = val，将当前元素与最后一个元素替换，并处理替换的元素。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
class Solution:
    def removeElement(self, nums, val):
        """
        :type nums: List[int]
        :type val: int
        :rtype: int
        """
        i=0
        n=len(nums)
        while i<n:
            if nums[i]==val:
                nums[i]=nums[n-1]
                n-=1#数组长度减1
            else:
                i+=1
        return n

时间复杂度：O(n)：在这种方法中，赋值操作的数量等于要删除的元素的数量。所以如果删除的元素很少，那么效率会更高。

空间复杂度：O(1)