移除元素(27)

45 阅读1分钟

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

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

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

解题思路:和前面26题思路几乎是一致的

  1. 定义两个指针 i 和 j,其中 j 表示当前不等于 val 的元素的末尾,i 表示当前需要查看是否等于 val 的元素
  2. 遍历数组,如果 arr[i] === val,则 i 向后移动,直到找到一个不等于 val 的元素
  3. 将找到的元素赋值给 arr[j],并将 i 和 j 都向后移动一位
  4. 重复步骤 2 和 3,直到遍历结束
  5. 返回 i 即可

下面是实现代码:

const removeElement = (arr, val) => {
  let len = arr.length;
  if (len === 0) return 0;
  let j = 0;
  for (let i = 0; i < len; i++) {
    if (arr[i] !== val) {
      // 将符合条件的元素放在前面
      arr[j] = arr[i];
      j++;
    }
  }
  return j;
};

console.log(removeElement([3, 2, 2, 3], 3));