删除排序数组中的重复项

Posted 2021-03-06 pigdragon

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

不要使用额外的数组空间，你必须在 原地 修改输入数组 并在使用 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。

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

解题思路：

解法： 双指针

首先注意数组是有序的，那么重复的元素一定会相邻。

要求删除重复元素，实际上就是将不重复的元素移到数组的左侧。

考虑用 2 个指针，一个在前记作 p，一个在后记作 q，算法流程如下：

1.比较 p 和 q 位置的元素是否相等。

如果相等，q 后移 1 位
如果不相等，将 q 位置的元素复制到 p+1 位置上，p 后移一位，q 后移 1 位
重复上述过程，直到 q 等于数组长度。

返回 p + 1，即为新数组长度。

画个图理解一下

 

 代码实现：

int removeDuplicates(int* nums, int numsSize){
    if(numsSize == 0 || nums == NULL){
        return 0;
    }
    int p=0;
    int q=1;
    while(q<numsSize){
        if(nums[p] == nums[q]){
            q++;
        }else{
            nums[p+1]=nums[q];
            p++;
            q++;
        }

    }
    return p+1;
}

精简之后的代码如下：

int removeDuplicates(int* nums, int numsSize){
    if(numsSize == 0 || nums == NULL){
        return 0;
    }
    int p=0;
    int q=1;
    while(q<numsSize){
        if(nums[p] !=nums[q]){
            nums[p+1]=nums[q];
            p++;
        }
         q++;

    }
    return p+1;
}

 

 

优化：

复杂度分析：

时间复杂度：O(n)O(n)O(n)。
空间复杂度：O(1)O(1)O(1)。

优化：

这是大部分题解都没有提出的，在这里提一下。

考虑如下数组：

此时数组中没有重复元素，按照上面的方法，每次比较时 nums[p] 都不等于 nums[q]，因此就会将 q 指向的元素原地复制一遍，这个操作其实是不必要的。

因此我们可以添加一个小判断，当 q - p > 1 时，才进行复制。

优化之后的代码：

int removeDuplicates(int* nums, int numsSize){
    if(numsSize == 0 || nums == NULL){
        return 0;
    }
    int p=0;
    int q=1;
    while(q<numsSize){
        if(nums[p] !=nums[q]){
            if(q-p >1){
                nums[p+1]=nums[q];
            }
            p++;
        }
         q++;

    }
    return p+1;
}



作者：max-LFszNScOfE
链接：https://leetcode-cn.com/problems/remove-duplicates-from-sorted-array/solution/shuang-zhi-zhen-shan-chu-zhong-fu-xiang-dai-you-hu/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/remove-duplicates-from-sorted-array
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。