快慢指针应用——Leetcode第80题删除有序数组中的重复项 II

给你一个有序数组 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 * 104
• -104 <= nums[i] <= 104
• nums 已按升序排列

这个题目和第26题非常相似，没刷到的小伙伴可以参考我上一篇博客。

写这题的时候我陷入了思维定式，以为可以和26题一样，只用快指针fast作为判断是否有重复项的标准，但是26题是只要求不可以重复，而80题要求不可以有两个以上的重复，这时沿用26题的方法就行不通，

if(nums[fast - 2] != nums[fast])

第一个测试用例的结果是nums = [ 1, 1, 2, 3 ] 。标准答案这边是：

if(nums[slow - 2] != nums[fast])

解决了会多删掉一个元素的问题。

下面附上代码和注释，请小伙伴们享用：

#include <iostream>
#include <vector>
using namespace std;

class Solution {
public: 
    int removeDuplicates(vector<int>& nums) {
        int n = nums.size();
        if(n <= 2)              // 如果vector长度小于等于2，直接返回长度
            return n;
        int fast = 2, slow = 2;
        while(fast < n){        // 循环直到快指针到达vector末尾
            if(nums[slow - 2] != nums[fast]){   // 如果慢指针前两个位置的值不等于快指针指向的值
                nums[slow] = nums[fast];        // 将快指针指向的值赋给慢指针指向的位置
                slow++;         // 慢指针右移
            }
            fast++;             // 快指针右移
        }
        return slow;            // 返回移除重复元素后的vector长度
    }
};

int main()
{
    Solution solution;
    vector<int> nums;
    cout << "Enter the elements of the vector (enter -1 to stop):" << endl;
    int num;
    while(true){
        cin >> num;
        if(num == -1)
            break;
        nums.push_back(num);
    }
    int result = solution.removeDuplicates(nums);
    cout << "The length of the new array is: " << result << endl;
    cout << "The number of the unique elements is: " << "[" ;
    for (int i = 0; i < result; i++) {
        cout << " " << nums[i] << " ";
    }
    cout << "]" << endl;
    return 0;
}

这道题整体思想和26题并没有变化，比较重要的点是要想清楚在有序数组中判断三个元素是否重复，只需要判断1、3两个位置的元素是否相同就行。