Leecode27. 移除元素

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

不要使用额外的数组空间，你必须仅使用 O(1) 额外空间并 原地 修改输入数组。

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

示例 1：

输入：nums = [3,2,2,3], val = 3
输出：2, nums = [2,2]

解释：函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。你不需要考虑数组中超出新长度后面的元素。例如，函数返回的新长度为2 ，而 nums = [2,2,3,3] 或 nums = [2,2,0,0]，也会被视作正确答案。

示例 2：

输入：nums = [0,1,2,2,3,0,4,2], val = 2
输出：5, nums = [0,1,4,0,3]
解释：函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。注意这五个元素可为任意顺序。你不需要考虑数组中超出新长度后面的元素。

提示：

  • 0 <= nums.length <= 100
  • 0 <= nums[i] <= 50
  • 0 <= val <= 100

习题解析：

因为数组元素在内存中是连续存储的，所以不能单独删除其中某个元素，只能选择覆盖方式，双重循环暴力解法可以解决此问题，但时间复杂度较高，为O(N^2)，所以可以采用双指针中的快慢指针来节省时间，快指针指向当前第一个不为目标元素的值（即如果执行覆盖操作第一个用于覆盖的值），慢指针指向需要第一个需要被覆盖的元素，然后快指针的值覆盖慢指针的值，然后一直向后遍历至快指针到达数组尾部，此时慢指针所指向的位置即为所有有效元素的长度。

习题代码：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
public int removeElement(int[] nums, int val) {

    int slow = 0,fast = 0;

    while (fast < nums.length){
      if (nums[fast] != val){
          nums[slow] = nums[fast];
          slow++;
      }
      fast++;
    }

    return slow;
}
  • 时间复杂度:O(N)
  • 空间复杂度:O(1)

习题链接

点我跳转习题