LeetCode #26 删除排序数组中的重复项

题目

给定一个排序数组，你需要在原地删除重复出现的元素，使得每个元素只出现一次，返回移除后数组的新长度。

不要使用额外的数组空间，你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。

示例 1:

给定数组 nums = [1,1,2], 

函数应该返回新的长度 2, 并且原数组 nums 的前两个元素被修改为 1, 2。 

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

示例 2:

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

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/remove-duplicates-from-sorted-array

双指针法

由于是排序后的数组，可以使用两个指针$i$和$j$。只要 $nums[i]=nums[j]$，便增加$j$跳过重复项。

当$nums[i]\ne nums[j]$时，已经完成跳过重复项，此时将$nums[j]$的值复制到$nums[i+1]$。然后递增$i$，接着再次重复相同的过程，直到$j$到达数组的末尾为止。

c++代码

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        int len = nums.size();
        if (len == 0) return 0;
        int i = 0;
        for(int j = 1;j < len;j++)
        {
            if(nums[i]!=nums[j])
            {
                i++;
                nums[i] = nums[j];
            }                               
        }
        return i+1;
    }
};

时间复杂度：$O(n)$，假设数组的长度是$n$，那么$i$和$j$分别最多遍历$n$步。

空间复杂度：$O(1)$。

优化

对于无重复数组，上述处理会导致不必要的复制，拖慢运行速度。因此，可加入一个判断：当$j-i>1$时，才复制。

优化代码

class Solution {
public:
    int removeDuplicates(vector<int>& nums) {
        int len = nums.size();
        if (len == 0) return 0;
        int i = 0;
        for(int j = 1;j < len;j++)
        {
            if(nums[i]!=nums[j])
            {
                i++;
                if(j - i > 0)	//由于上一行已经使用i++，故此时条件应为j - (i-1) > 1，即j - i > 0
                	nums[i] = nums[j];
            }                               
        }
        return i+1;
    }
};