LeetCode 27. 移除元素(JavaScript 简单)

1. 题目

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

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

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

说明:

为什么返回数值是整数,但输出的答案是数组呢?

请注意,输入数组是以**「引用」**方式传递的,这意味着在函数里修改输入数组对于调用者是可见的。

你可以想象内部操作如下:

复制代码
// nums 是以“引用”方式传递的。也就是说,不对实参作任何拷贝
int len = removeElement(nums, val);

// 在函数里修改输入数组对于调用者是可见的。
// 根据你的函数返回的长度, 它会打印出数组中 该长度范围内 的所有元素。
for (int i = 0; i < len; i++) {
    print(nums[i]);
}

示例 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

2. 思路及代码

2.1 原思路和代码

利用双指针,一个指针p1指向要被替代的位置(此位置的值为 val),另一指针p2指向非val值nums[p2]。让 p2 指向的值替换掉 p1 位置上的原值。

注意点:

  • p2 要保持在 p1 之后
  • 由于 JavaScript 的数组不存在越界,要时常注意判断 p1 和 p2 是否越界了
  • 另外最终返回数组的长度,所以要搞清楚返回的边界是 p1 还是 p1-1
javascript 复制代码
/**
 * @param {number[]} nums
 * @param {number} val
 * @return {number}
 */
var removeElement = function(nums, val) {
    let p1=0, p2=0; //p1指向当前可能要被替换的位置,p2用遍历数组找到应该存下来的值
    n = nums.length
    while(p2<n && p1<n){
        while(p1<n && nums[p1]!=val){
            p1++; //找到当前值为 val 的位置 p1
        } 
        for(p2=p1+1; nums[p2]==val; p2++); //找到当前值不为 val 的值 nums[p2]
        if(p2>=n || p1>=n) break;
        nums[p1] = nums[p2];
        nums[p2] = val;
        p1++;
        p2++;
    }
    return p1;
};

2.2 反思和升级代码

  • 判断+移动 是两次计算,如果直接 移动 可以解决的,额外再判断一次浪费时间。所以可以不需要判断 p2 指向的值是不是 val。如果 p2 指向的值是 val,直接复制过来,只要 当 p1 指向的值不是 val 时 p1 才动即可。
  • 如果 p1 从数组起点开始,p2 从数组末尾开始,循环直到 p1===p2,这样总共循环只有 n(数组长度) 次。
  • 将 p2 位置上的值复制到 p1 位置上时注意。如果保证 p2 当前指向的值已经被复制到 p1 上(先挪动再复制),则最终 p1 和 p2 重合时,++此时重合位置已经处理过++,直接返回的p1就是数组长度
    • 如果是 p1 挪到 p2 上,p2指向的位置已被处理,则p1,p2此时指向的元素已经被复制到前面位置上,此时p1的下标就是数组长度;
    • 如果是 p2 挪到 p1 上,此时p2指向的位置已经被处理,p1和p2指向的位置就是上一步自己复制给自己的操作,nums[left]===val,当前元素是val,返回的p1除了自己当前的位置之前的所有位置都不是val,返回的 p1 就是数组长度。
  • p2 指向位置的值可能还没有被复制到 p1 上,当 p1 与 p2 重合时,还需要额外再讨论一次,当前 p1 指向的值是 val 时,返回的数组长度为 p1;当前 p1 指向的值不是 val 时,返回数组的长度应该是 p1+1。

双指针优化

left 指针(规定, 如果需要两个指针且存在 p1<p2 的循环结束条件时,命名为leftright)从数组索引 0 开始遍历,right 指针从数组索引 num.length()-1 开始遍历。当 nums[left] === val 时,将 nums[right - 1] 的值复制过来,right--;当 nums[left] !== val 时,left++。重复上句操作,直到 left===right 时循环终止,返回 left。

javascript 复制代码
/**
 * @param {number[]} nums
 * @param {number} val
 * @return {number}
 */
var removeElement = function(nums, val) {
    let left=0, right=nums.length; //left指向当前可能要被替换的位置,right用遍历数组找到应该存下来的值
    while(left < right){
        if(nums[left] == val) nums[left] = nums[--right];
        else left++;
    }
    return left
};

3. 总结经验

  1. 能不判断解决问题的,不要判断。获得最好的时间复杂度,不需要过细考虑。
  2. 在块作用域内起作用的变量,使用 let。
相关推荐
Captain823Jack1 小时前
nlp新词发现——浅析 TF·IDF
人工智能·python·深度学习·神经网络·算法·自然语言处理
Captain823Jack2 小时前
w04_nlp大模型训练·中文分词
人工智能·python·深度学习·神经网络·算法·自然语言处理·中文分词
是小胡嘛2 小时前
数据结构之旅:红黑树如何驱动 Set 和 Map
数据结构·算法
m0_748255022 小时前
前端常用算法集合
前端·算法
真的很上进2 小时前
如何借助 Babel+TS+ESLint 构建现代 JS 工程环境?
java·前端·javascript·css·react.js·vue·html
呆呆的猫3 小时前
【LeetCode】227、基本计算器 II
算法·leetcode·职场和发展
Tisfy3 小时前
LeetCode 1705.吃苹果的最大数目:贪心(优先队列) - 清晰题解
算法·leetcode·优先队列·贪心·
余额不足121383 小时前
C语言基础十六:枚举、c语言中文件的读写操作
linux·c语言·算法
火星机器人life5 小时前
基于ceres优化的3d激光雷达开源算法
算法·3d
虽千万人 吾往矣5 小时前
golang LeetCode 热题 100(动态规划)-更新中
算法·leetcode·动态规划