leetcode80.删除排序数组中的重复项 II

leetcode80.删除排序数组中的重复项 II

题目描述

给定一个排序数组，你需要在原地删除重复出现的元素，使得每个元素最多出现两次，返回移除后数组的新长度。
不要使用额外的数组空间，你必须在原地修改输入数组并在使用 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 。
你不需要考虑数组中超出新长度后面的元素。

解题思路

此题可以使用暴力求解的方法，在遇到重复的元素数量大于2的时候，使用两层循环不断得使后面的每一个元素向前移动，时间复杂度O(n²)

这里使用一种双指针的方法。对于一个给定的数组，如果数组的长度小于等于2的话，就直接返回数组的长度。当长度大于2的时候，定义两个指针，其中一个指针用于移动数组下标，另一个指针index用于移动修改后数组的下标。两个指针的移动规则，移动数组的指针可以用循环代替，index指针，当重复的元素小于2时，index向前移动，并且与当前位置元素交换，当数量大于2时，不移动，下面来一个动画说明一下：

题目要求返回值是修改后的长度，不是下标，所以需要index+1

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        if(nums.size() <= 2){
            return nums.size();
        }
        
        int index = -1;
        int temp = nums[0];
        int count = 0;
        
        for(int i=0; i<nums.size(); i++){
            if(nums[i] == temp){
                count++;
                if(count <= 2){  // 如果相等，在count小于等于2的时候，index才移动
                    mySwap(nums[i], nums[index+1]);
                    index++;
                }
            }
            else{
                temp = nums[i];      // 不相等index和当前位置元素做交换
                count = 1;
                mySwap(nums[i], nums[index+1]);   
                index++;
            }
        }
        
        return index+1;
    }
    
    // 交换两个数
    void mySwap(int& a, int& b){
        int temp = a;
        b = a;
        a = temp;
    }
};

欢迎大家关注我的个人公众号，同样的也是和该博客账号一样，专注分享技术问题，我们一起学习进步