力扣NO80. 删除排序数组中的重复项 II的C++双指针法

力扣NO.80删除排序数组中的重复项 II

给定一个排序数组，你需要在原地删除重复出现的元素，使得每个元素最多出现两次，返回移除后数组的新长度。不要使用额外的数组空间，你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。

示例 1:

给定 nums = [1,1,1,2,2,3],

函数应返回新长度 length = 5, 并且原数组的前五个元素被修改为 1, 1, 2, 2, 3

不需要考虑数组中超出新长度后面的元素。

思路：
利用两个指针 i 和 j ，i 指针用来遍历数组， j 指针记录重复<=2元素的位置，并替换原数组的元素。

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        // 判断是否为空
        if (nums.empty()) {  
            return 0;
        }
        int i = 0, j = 0, k = 0;  // i指针遍历数组，j指针记录重复<=2元素的位置
        int a = nums[0];  // 记录当前重复元素的值
        while (i < nums.size()) {   
            if (a == nums[i]) {     // 与当前重复元素相等--标签k自增
                k += 1;
                if (k <= 2) {    // 标签值小于等于2时， 指针j记录位置
                    j += 1;
                    nums[j - 1] = nums[i];  // 在原数组中进行替换
                }
            } else {     // 与当前值不等--替换当前记录值
                k = 1;
                j += 1;
                nums[j - 1] = nums[i];
                a = nums[i];
            }
            i += 1;    // 继续遍历数组
        }
        return j;    // 返回去除数组的长度
    }
};

作者：Microimmortal
链接：https://leetcode-cn.com/problems/remove-duplicates-from-sorted-array-ii/solution/cshuang-zhi-zhen-by-microimmortal/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。