LeetCode80 RemoveDuplicatedFromSortedArrayⅡ

LeetCode80 RemoveDuplicatedFromSortedArrayⅡ

题目描述

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

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

示例 1:

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

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

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

难度：medium

解答

是LeetCode26的升级版，同样需要用双指针法，一次遍历，慢指针为 k，快指针 i，计数器count。

数组从[0,k]为最多重复2次的数组元素，i用于去遍历数组，count用以计算nums[k]的值出现的次数，为1或者2.

首先nums[i]!=nums[k]时，说明是第一次遇见这个数，于是放到k++的位置上，此时nums[k]的count为1。

当nums[i]==nums[k]时，如果此时的count为1，说明是第二次出现，可以放入k++的位置上，count++；当count不为1时，说明是多次出现（本题目中是2次以上）则直接跳过。

因为最后得到的数组是从0到k，即[0,k]，所以返回k+1。

public static int solve1(int[] nums){
        int count=1;
        if(nums.length==0) return 0;
        int k=0;
        for(int i=1;i<nums.length;i++){
            if(nums[i]!=nums[k]) {
                k++;
                nums[k]=nums[i];
                count=1;
            }else if(nums[i]==nums[k]  && count==1 ){
                k++;
                nums[k]=nums[i];
                count++;
            }
        }
        return k+1;
    }

注意：本段代码中需要快慢指针的初始化，k=0, i=1，用以确保nums[i]==nums[k]指的不是同一个元素，否则若是设置i=0，则第二个if判断需要再加上一个条件 i!=k。

另外，此题目可以扩展为每个元素最多出现maxCount次。只需要更改条件count<maxCount

public static int solve1(int[] nums, int maxcount){
    int count=1;
    if(nums.length==0) return 0;
    int k=0;
    for(int i=1;i<nums.length;i++){
        if(nums[i]!=nums[k]) {
            k++;
            nums[k]=nums[i];
            count=1;
        }else if(nums[i]==nums[k]  && count<maxcount ){
            k++;
            nums[k]=nums[i];
            count++;
        }
    }
    return k+1;
}