LeetCode —— 26.删除排序数组中的重复项

183 阅读4分钟

我报名参加金石计划1期挑战——瓜分10万奖池,这是我的第2篇文章,点击查看活动详情

该题是数组移除元素题型第二题。

题目来源

26. 删除有序数组中的重复项(LeetCode)

题目描述(简单

给你一个 升序排列 的数组 nums ,请你 原地 删除重复出现的元素,使每个元素 只出现一次 ,返回删除后数组的新长度。元素的 相对顺序 应该保持 一致

由于在某些语言中不能改变数组的长度,所以必须将结果放在数组 nums 的第一部分。更规范地说,如果在删除重复项之后有 k 个元素,那么 nums 的前 k 个元素应该保存最终结果。

将最终结果插入 nums 的前 k 个位置后返回 k

不要使用额外的空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。

判题标准:

系统会用下面的代码来测试你的题解:

int[] nums = [...]; // 输入数组
int[] expectedNums = [...]; // 长度正确的期望答案

int k = removeDuplicates(nums); // 调用

assert k == expectedNums.length;
for (int i = 0; i < k; i++) {
    assert nums[i] == expectedNums[i];
}

如果所有断言都通过,那么您的题解将被 通过

示例1

输入:nums = [1,1,2]
输出:2, nums = [1,2,_]
解释:函数应该返回新的长度 2 ,并且原数组 nums 的前两个元素被修改为 1, 2 。不需要考虑数组中超出新长度后面的元素。

示例2

输入:nums = [0,0,1,1,1,2,2,3,3,4]
输出:5, nums = [0,1,2,3,4]
解释:函数应该返回新的长度 5 , 并且原数组 nums 的前五个元素被修改为 0, 1, 2, 3, 4 。不需要考虑数组中超出新长度后面的元素。

提示:

  • 1 <= nums.length <= 3 * 104
  • -104 <= nums[i] <= 104
  • nums 已按 升序 排列

题目解析:

这道题目的要求是:对给定的有序数组 nums 删除重复元素,在删除重复元素之后,每个元素只出现一次,并返回新的长度。

方法一:暴力删除

只需遍历数组,判断前后两项的值是否相等,如若相等便将相等项删除一个,但是由于删除时会改变数组长度,于是对于遍历的变量需要减一,保证遍历到了所有数据。

代码(JavaScript)

/**
 * @param {number[]} nums
 * @return {number}
 */
var removeDuplicates = function(nums) {
    for(let i=0;i<nums.length;i++){
        if(nums[i]===nums[i+1]){
            //删除相等项
            nums.splice(i+1,1)
            i--
        }
    }
};

如图:

image.png

方法二:双指针

不难发现,在有序数组中,重复项必定是相连的,于是我们采用双指针的方法删除重复项。根据题意我们只需保证前 n 项为该数组中所有的不同项。于是我们可以定义slow和fast两个指针,分别代表慢指针(表示下一个不同元素的位置)和快指针(标识遍历数组的下标位置)。

代码(JavaScript)

/**
 * @param {number[]} nums
 * @return {number}
 */
var removeDuplicates = function(nums) {
    let slow=0
    for(let fast=0;fast<nums.length;fast++){
        if(nums[fast]!==nums[fast+1]){
            nums[slow]=nums[fast]
            slow++
        }
    }
    nums.splice(slow,nums.length-slow)
};

如图:

image.png

我一看不对劲呀,然后再次查看判断标准。原来只需修改数组的值的排序,不必删除数组元素,即该数组共有 n 个不重复的元素,我只需保证该数组的前 n 项为 n 个不重复元素即可,所以在循环结束后,我返回了不重复元素的个数,正好满足断言( assert k == expectedNums.length; )要求。

优化

直接返回数组的前 n 项,不对数组的总长度进行更改。

/**
 * @param {number[]} nums
 * @return {number}
 */
var removeDuplicates = function (nums) {
    let slow = 1
    for (let fast = 0; fast < nums.length; fast++) {
        if (nums[fast] === nums[slow - 1]) {
            continue
        }
        nums[slow++] = nums[fast]
    }
    return slow
};

如图:

image.png

优化二

由于 while 执行循环体的速度会比 for 快,于是进行了以下修改。

/**
 * @param {number[]} nums
 * @return {number}
 */
var removeDuplicates = function (nums) {
    let slow = 0
    let fast = 0
    while (fast < nums.length) {
        if (nums[slow] !== nums[fast]) {
            nums[++slow] = nums[fast]
        }
        fast++
    }
    return slow + 1
};

如图:

image.png

image.png

参考

执行用时和内存消耗仅供参考,大家可以多提交几次。如有更好的想法,欢迎大家提出。