力扣LeetCode: 80 删除有序数组中的重复项Ⅱ

不想编程小谭

于 2025-02-09 22:16:19 发布

阅读量963

点赞数 18

分类专栏： LeetCode 文章标签： leetcode 算法数据结构

本文链接：https://blog.youkuaiyun.com/weixin_73982851/article/details/145538044

版权

LeetCode 专栏收录该内容

42 篇文章

订阅专栏

题目：

给你一个有序数组 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 已按升序排列

解法一：双指针

解决思路

有序数组的特性：
- 因为数组是有序的，所以重复的元素一定是连续的。
- 我们需要确保每个元素最多出现两次。
双指针法：
- 使用两个指针：slow 和 fast。
- slow 指针用于构建新数组，指向当前可以放置新元素的位置。
- fast 指针用于遍历原数组，检查每个元素是否可以保留。
核心逻辑：
- 如果当前元素 nums[fast] 不等于 nums[slow - 2]，说明 nums[fast] 可以保留，并且不会导致重复超过两次。
- 将 nums[fast] 放到 nums[slow] 的位置，然后移动 slow 指针。
- 如果 nums[fast] 等于 nums[slow - 2]，说明 nums[fast] 已经出现了两次，跳过它。
边界条件：
- 如果数组长度小于等于 2，直接返回原数组长度，因为不可能有重复超过两次的元素。

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        int n = nums.size();
        if (n <= 2) return n; // 如果数组长度小于等于2，直接返回

n = nums.size()：获取数组的长度。
if (n <= 2) return n;：如果数组长度小于等于 2，直接返回原数组长度，因为不可能有重复超过两次的元素。

        int slow = 2; // 慢指针从索引2开始
        for (int fast = 2; fast < n; ++fast) {

slow = 2：慢指针从索引 2 开始，因为前两个元素无论如何都可以保留。
for (int fast = 2; fast < n; ++fast)：快指针从索引 2 开始遍历数组。

            // 如果当前元素不等于慢指针前两个元素
            if (nums[fast] != nums[slow - 2]) {
                nums[slow] = nums[fast]; // 将当前元素放到慢指针位置
                ++slow; // 移动慢指针
            }

if (nums[fast] != nums[slow - 2])：
- 检查当前元素 nums[fast] 是否等于 nums[slow - 2]。
- 如果不等，说明 nums[fast] 可以保留，并且不会导致重复超过两次。
nums[slow] = nums[fast];：
- 将 nums[fast] 放到 nums[slow] 的位置。
++slow;：
- 移动慢指针，指向下一个可以放置新元素的位置。

        }
        return slow; // 返回新数组的长度
    }
};

return slow;：
- 最终 slow 指针的位置就是新数组的长度。

完整代码

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        int n = nums.size();
        if (n <= 2) return n; // 如果数组长度小于等于2，直接返回

        int slow = 2; // 慢指针从索引2开始
        for (int fast = 2; fast < n; ++fast) {
            // 如果当前元素不等于慢指针前两个元素
            if (nums[fast] != nums[slow - 2]) {
                nums[slow] = nums[fast]; // 将当前元素放到慢指针位置
                ++slow; // 移动慢指针
            }
        }
        return slow; // 返回新数组的长度
    }
};

解法二：计数器遍历

遍历数组，使用一个计数器来记录当前元素的出现次数。
如果当前元素与前一个元素相同，则增加计数器。
如果当前元素与前一个元素不同，则重置计数器。
如果计数器的值小于等于2，则保留该元素，否则跳过该元素。

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        if (nums.size() <= 2) {
            return nums.size();
        }

        int index = 1;  // 从第二个元素开始
        int count = 1;  // 计数器，记录当前元素的出现次数

        for (int i = 1; i < nums.size(); ++i) {
            if (nums[i] == nums[i - 1]) {
                count++;
            } else {
                count = 1;  // 重置计数器
            }

            if (count <= 2) {
                nums[index++] = nums[i];
            }
        }

        return index;
    }
};