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

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/qq_35356190/article/details/82913319

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

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

你不需要考虑数组中超出新长度后面的元素。
class Solution {
    public int removeDuplicates(int[] nums) {
        if(null == nums){
            return 0;
        }
        if(nums.length<3){
            return nums.length;
        }
        int repeatCount = 1, count = 0;
        for(int i=nums.length-1;i>0;--i){
            if(nums[i]==nums[i-1]){
                if(1==repeatCount){
                    ++repeatCount;
                }else if(2==repeatCount){
                    for(int j=i+1;j<nums.length;++j){
                        nums[j-1]=nums[j];
                    }
                    ++count;
                }
            }else{
                repeatCount=1;
            }
        }
        return nums.length-count;
    }
}

刚开始思路不是很清晰,变量i从0遍历到数组尾部,answer error!
因为会重复数为2时,数组向前移动一位,会覆盖索引i处,覆盖原来的值,然后i继续向前遍历,会导致i值变化,而repeatCount=2,继续查训会导致repeatCount不准确。

错误例:

[1][1][1][2][2][3]
[1][2][2][2][3]
[1][1][2][3]

逆向思维,从后往前遍历,则可以避免覆盖情形出现。

二 O(n)解法

class Solution {
    public int removeDuplicates(int[] nums) {
        if(null == nums){
            return 0;
        }
        if(nums.length<3){
            return nums.length;
        }
        int i=1;
        for(int j=i+1;j<nums.length;++j){
            if(nums[i]!=nums[j] || (nums[j]==nums[i] && nums[i-1]!=nums[j])){
                nums[i+1]=nums[j];
                ++i;
            }
        }
        return i+1;
    }
}

目标:一次过!最优解!

猜你喜欢

转载自blog.csdn.net/qq_35356190/article/details/82913319