LeetCode移除元素

题目描述

题目要求给你一个数组nums和一个值val，需要原地移除数组中所有等于val的元素，并返回移除后数组的新长度。算法需要满足以下条件：

• 不使用额外的数组空间，即空间复杂度必须是O(1)。
• 原地修改输入数组。
• 数组元素的相对顺序可以改变。

示例

• 示例1: nums = [3,2,2,3], val = 3，返回新长度2，数组变为[2,2]。
• 示例2: nums = [0,1,2,2,3,0,4,2], val = 2，返回新长度5，数组变为[0,1,3,0,4]。

思路分析

由于数组元素在内存中是连续的，不能单独删除数组中的某个元素，只能通过覆盖的方式进行操作。

暴力解法

暴力解法使用两层循环，外层循环遍历数组，内层循环将发现的等于val的元素删除，时间复杂度为O(n^2)。

// 时间复杂度：O(n^2)
// 空间复杂度：O(1)
class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        int size = nums.size();
        for (int i = 0; i < size; i++) {
            if (nums[i] == val) { // 发现需要移除的元素，就将数组集体向前移动一位
                for (int j = i + 1; j < size; j++) {
                    nums[j - 1] = nums[j];
                }
                i--; // 因为下标i以后的数值都向前移动了一位，所以i也向前移动一位
                size--; // 此时数组的大小-1
            }
        }
        return size;

    }
};

双指针法

双指针法是一种更高效的解决方案，时间复杂度为O(n)：

• 使用两个指针，一个快指针fastIndex用来遍历数组，一个慢指针slowIndex指向新数组的下标位置。
• 快指针fastIndex从头开始遍历数组，如果发现nums[fastIndex]不等于val，则将其复制到slowIndex的位置，并将slowIndex向前移动一位。
• 继续移动快指针，重复上述操作，直到遍历完整个数组。

• class Solution {
      public int removeElement(int[] nums, int val) {
          // 快慢指针
          int slowIndex = 0;
          for (int fastIndex = 0; fastIndex < nums.length; fastIndex++) {
              if (nums[fastIndex] != val) {
                  nums[slowIndex] = nums[fastIndex];
                  slowIndex++;
              }
          }
          return slowIndex;
      }
  }