欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

LeetCode刷题之删除排序数组中的重复项

程序员文章站 2022-05-20 11:20:43
...

LeetCode刷题之删除排序数组中的重复项

题目

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

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

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

说明:

为什么返回数值是整数,但输出的答案是数组呢?

请注意,输入数组是以“引用”方式传递的,这意味着在函数里修改输入数组对于调用者是可见的。

你可以想象内部操作如下:

nums 是以“引用”方式传递的。也就是说,不对实参做任何拷贝
int len = removeDuplicates(nums);

在函数里修改输入数组对于调用者是可见的。
根据你的函数返回的长度, 它会打印出数组中该长度范围内的所有元素。
for (int i = 0; i < len; i++) {
    print(nums[i]);
}

题解

首先我们注意到题目里面要求是有序数组,既然是有序数组,那么其中
重复的元素一定会相邻。

其次,要求原地修改。

我们可以使用双指针,一个在前记作 p,一个在后记作 q,直接在原数组上进行操作。比较 p 和 q 位置的元素是否相等。如果相等,q 后移 1 位。如果不相等,将 q 位置的元素复制到 p+1 位置上,p 后移一位,q 后移 1 位。

重复上述过程,直到 q 等于数组长度。

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

最后注意:

1.数组不为空或数组长度不为0。

2.q大于数组长度时,不再进行比较。

class Solution {
    public int removeDuplicates(int[] nums) {
         if (nums.length == 0) return 0;
    int p = 0;
    for (int q = 1; q < nums.length; q++) {
        if (nums[q] != nums[p]) {
            p++;
            nums[p] = nums[q];
        }
    }
    return p + 1;
}
}

复杂度分析

时间复杂度:O(n),假设数组的长度是 n,两个指针最多遍历n步。

空间复杂度:O(1)。