美文网首页数据结构和算法
算法练习27:移除元素(leetcode 27)

算法练习27:移除元素(leetcode 27)

作者: miao8862 | 来源:发表于2021-05-27 19:37 被阅读0次

题目

给你一个数组 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],也会被视作正确答案。

解法

使用双指针,左指针start指向数组nums头位置,右指针end指向数组尾位置。

  • start指向的数等于val,即nums[start] === val时,两指针的数交换,右指针左移一位end--,代表最右一位是重复的数,不再需要
  • 如果不start指向的数不等于val时,代表start指向的数,最后要保留的数,左指针右移一位start++
  • 当左指针和右指针相遇时,代表循环结束,此时判断当前位置的数是否等于val,如果等于,代表当前位的数是不要的,此时左边个数为start个;否则加上当前位数,start + 1
  • 注意边界值,当数组为空时,不需要遍历,直接返回0
/**
 * @param {number[]} nums
 * @param {number} val
 * @return {number}
 */
var removeElement = function(nums, val) {
    if(!nums.length) return 0;
    let start = 0;
    let end = nums.length - 1;
    while(start !== end) {
        if(nums[start] === val) {
            [nums[start], nums[end]] = [nums[end], nums[start]]
            end--;
        }else {
            start++;
        }
    }
    if(nums[start] === val) {
        return start;
    }
    return start + 1;
};

优化一下

因为题目最后只要位数,所以其实并不需要左右指针的值交换,只需要把右指针的数移到左指针的数就可以了

var removeElement = function(nums, val) {
    if(!nums.length) return 0;
    let start = 0;
    let end = nums.length - 1;
    while(start !== end) {
        if(nums[start] === val) {
            // 不用交换,只需要把右指针的数移到左指针上即可
            nums[start] = nums[end]
            end--;
        }else {
            start++;
        }
    }
    if(nums[start] === val) {
        return start;
    }
    return start + 1;
};
  • 时间复杂度:O(n)
  • 空间复杂度:O(1)

相关文章

网友评论

    本文标题:算法练习27:移除元素(leetcode 27)

    本文链接:https://www.haomeiwen.com/subject/rwpdsltx.html