LeetCode刷题_80:删除排序数组中的重复项 II【C++】-优快云博客

本文链接：https://blog.youkuaiyun.com/anchuanxu/article/details/98226530

题目描述

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

示例 1:

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

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

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

示例 2:

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

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

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

答案代码

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {

        //首先判断数组是否为空
        if(nums.empty()) return 0;
        //当数组元素小于等于2的时候，直接返回，不做操作
        if(nums.size() <= 2) return nums.size();

        int k = 0;//指针1
        int count = 0;//计数器，当计数器的值>2的时候，k停止移动

        //开始扫描数组
        for(int i = 1;i < nums.size();i++){
            if(nums[i] == nums[k]){ // 两者元素相等，计算器加1
                count++;
                if(count < 2){ //计数器小于2的时候，K向前移动
                    k++;
                nums[k] = nums[i];} // 赋值，就算相等也要赋值，不然有连续重复元素的时候，数据会出现错误。
            }
            else
            {
               nums[++k] = nums[i]; //不等的时候，先移动K指针，然后赋值，再置计数器0；
                count=0;
            }
        }
        return k+1; 返回数组长度，为指针K+1；
    }
};

测试代码

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

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {

        //首先判断数组是否为空
        if(nums.empty()) return 0;
        //当数组元素小于等于2的时候，直接返回，不做操作
        if(nums.size() <= 2) return nums.size();

        int k = 0;//指针1
        int count = 0;//计数器，当计数器的值>2的时候，k停止移动

        //开始扫描数组
        for(int i = 1;i < nums.size();i++){
            if(nums[i] == nums[k]){
                count++;
                if(count < 2){
                    k++;
                nums[k] = nums[i];}
            }
            else
            {
               nums[++k] = nums[i];
                count=0;
            }
        }
        return k+1;
    }
};

int main() {
    int arr[] = {0,0,1,1,2,2};
    int brr[] = {0,0,1,1,1,2,2};
    int crr[] = {0,0,1,1,2,2,2,2};

    vector<int> vec_a(arr,arr+sizeof(arr)/sizeof(int));
    vector<int> vec_b(brr,brr+sizeof(brr)/sizeof(int));
    vector<int> vec_c(crr,crr+sizeof(crr)/sizeof(int));

    cout<<Solution().removeDuplicates(vec_a)<<endl;
    cout<<Solution().removeDuplicates(vec_b)<<endl;
    cout<<Solution().removeDuplicates(vec_c)<<endl;

    return 0;
}