前端算法(27)

69 阅读1分钟

本文已参与「新人创作礼」活动,一起开启掘金创作之路。

题目

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

不要使用额外的数组空间,你必须仅使用 O(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],也会被视作正确答案。

题目解析

思路一

我们可以利用双指针,先将 nums 中等于 val 的元素移动到数组末尾,当 left=right 时, 如果 nums[left]等于 val, 即删除 nums[left], 返回 left 的值即可;如果 nums[left]不等于 val, 即不删除 nums[left], 返回 left+1

var removeElement = function (nums, val) {
  if (!nums.length) return 0;
  let left = 0,right = nums.length - 1;
  while (left < right) {
    // 比较val和左指针指向的元素
    while (left < right && nums[left] !== val) {
      left++;
    }
    // 比较val和右指针指向的元素
    while (left < right && nums[right] === val) {
      right--;
    }
    // 跳出内部循环时, nums[left] === val、nums[right] !== val
    // 交换nums[left]和nums[right],此时数组尾部全是值为val的元素
    [nums[left], nums[right]] = [nums[right], nums[left]];
  }
  // 判断left===right时,nums[left]和val的关系
  return nums[left] === val ? left : left + 1;
};

思路二

循环nums数组,如果当前值不等于val,则nums[m++]进行赋值当前值,顺便m+1进行下一次对比时赋值,最后返回m(不需要加1,因为m++的++在后面,结束循环时会比循环时大1)

var removeElement = function (nums, val) {
    const len = nums.length;
    if (len === 0) return 0;
    let m = 0;
    for (let i = 0; i < len; i++) {
        if (nums[i] !== val) {
            nums[m++] = nums[i];
        }
    }
    return m;
};