「LeetCode」80.删除有序数组中的重复项 II「这是我参与2022首次更文挑战的第26天，活动详情查看：2022

「这是我参与2022首次更文挑战的第26天，活动详情查看：2022首次更文挑战」。

题目描述🌍

给你一个有序数组 nums ，请你原地删除重复出现的元素，使每个元素 最多出现两次 ，返回删除后数组的新长度。

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

说明：

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

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

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

// nums 是以“引用”方式传递的。也就是说，不对实参做任何拷贝
int len = removeDuplicates(nums);

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

示例 1：

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

示例 2：

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

提示：

$1$ <= nums.length <= $3 * 10^4$
$-10^4$ <= nums[i] <= $10^4$
nums 已按升序排列

双指针法🚀

解题思路

这题是「LeetCode」26.删除有序数组中的重复项的拓展题，所以我们可以尝试借助那题的解题思路。

还是使用双指针，指针 index 指向有效数组末尾的后一位，指针 i 用于遍历数组。

因为每个元素最多出现 2 次，那对于当前遍历指针指向的元素 nums[i] 而言，如果与有效数组倒数第 2 位元素不同，那么就用当前元素覆盖指针 index 指向的位置，且让 index 指针后移。

图 1

图 2

😎边界值分析（3次、2次、1次）

因为数组是有序的，所以只要 nums[i] == nums[index-2]，就说明了 nums[index-1] 和这两个元素都是同一个值（夹逼准则），这样就存在至少 3 个相同连续的元素；
若 nums[i] != nums[index-2]，那尽管 nums[index-1] 和当前元素 nums[i] 相同，也是刚好符合题目最多 2 次的要求；
若 nums[i] != nums[index-2]，且有效数组最后一位 nums[index-1] 和 nums[i] 不同，那更无关紧要了，因为这两连续元素都只出现 1 次。

代码

Java

class Solution {
    public int removeDuplicates(int[] nums) {
        int index = 0;
        for (int i = 0; i < nums.length; i++) {
            if (index < 2 || nums[index - 2] != nums[i])
                nums[index++] = nums[i];
        }
        return index;
    }
}

C++

class Solution {
public:
    int removeDuplicates(vector<int> &nums) {
        int index = 0;
        int length = nums.size();
        for (int i = 0; i < length; i++) {
            if (index < 2 || nums[index - 2] != nums[i])
                nums[index++] = nums[i];
        }
        return index;
    }
};

时间复杂度： $O(n)$

空间复杂度： $O(1)$

扩展🌓

「通用模板」

回顾一下三叶姐归纳的「通用模板」🤩

Java

class Solution {
    public int removeDuplicates(int[] nums, int k) {
        int index = 0;
        for (int num : nums) {
            if (index < k || nums[index - k] != num)
                nums[index++] = num;
        }
        return index;
    }
}

C++

class Solution {
public:
    int removeDuplicates(vector<int> &nums) {
        return process(nums, 1);
    }

    int process(vector<int> &nums, int k) {
        int idx = 0;
        for (auto x: nums) {
            if (idx < k || nums[idx - k] != x) {
                nums[idx++] = x;
            }
        }
        return idx;
    }
};

最后🌅

该篇文章为 「LeetCode」 系列的 No.24 篇，在这个系列文章中：

尽量给出多种解题思路
提供题解的多语言代码实现
记录该题涉及的知识点

👨‍💻争取做到 LeetCode 每日 1 题，所有的题解/代码均收录于「LeetCode」仓库，欢迎随时加入我们的刷题小分队！