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

2019/01/10 23:35 下午 posted in  leetcode

问题

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

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

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

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

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

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

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

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

解法

使用快慢指针,快指针每次循环都往前推进,慢指针只有在指针位置小于2(前面两位必然需要向前推进)或者慢指针前推2位的数小于快指针的数时,才向前推进,并且将快指针的值赋值到慢指针位置。

源码

class Solution {
    public int removeDuplicates(int[] nums) {
        int fast=0;
        int slow=0;
        while(fast<nums.length) {
            //慢指针只有在小于2或者前数2位小于快指针时才往前推进
            if (slow < 2 || nums[fast] > nums[slow-2]) {
                nums[slow] = nums[fast];
                slow++;
            }
            fast++;
        }
        return slow;
    }
}