[算法] leetcode 26. 删除有序数组中的重复项

46 阅读2分钟

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

题目

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

考虑 nums 的唯一元素的数量为 k ,你需要做以下事情确保你的题解可以被通过:

更改数组 nums ,使 nums 的前 k 个元素包含唯一元素,并按照它们最初在 nums 中出现的顺序排> 列。nums 的其余元素与 nums 的大小不重要。

返回 k 。

说明

为判题标准:

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

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 。不需要考虑数组中超出新长度后面的元素。

思路分析

  • 不要使用额外的数组空间 说明需要使用双指针
  • 相对顺序需要保持一致, 说明我们需要从头遍历

伪代码

  • i: 唯一元素的最大index;j: 当前遍历比较的元素

  • i初始值为0

  • j从第二个元素开始, 比较与 nums[i] 元素是否相同

    • 相同,则忽略 j++
    • 不同,则将i的下一位设置成当前 nums[j], j++
    • 直到遍历结束
  • 返回i+1 作为结果(结果要的是个数)

边界条件

  • 遍历完数组结束, 即j >= nums.length

难点

  • 由于j的初始值为1(即指向第二个元素), 所以需要考虑当数组长度为1
    • 当数组长度为1时, 则说明当前数组已经符合规定
    • 说明我们无需遍历, 此时我们的j = 1 == nums.length, 直接返回,
    • 即此时返回 1 (i + 1) 是正确的

代码

class Solution {
    public int removeDuplicates(int[] nums) {
        int l = nums.length;
        int pInd = 0;
        int cInd = 1;
        while(cInd < l) {
            if(nums[cInd] != nums[pInd]) {
                nums[++pInd] = nums[cInd];
            }
            cInd++;
        }
        return pInd + 1;
    }
}

结果