第14&15题 Remove Duplicates from Sorted Array I&II

最新推荐文章于 2023-09-22 13:35:40 发布

原创最新推荐文章于 2023-09-22 13:35:40 发布 · 648 阅读

0 ·

CC 4.0 BY-SA版权

CodeReview 同时被 2 个专栏收录

76 篇文章

订阅专栏

CodeReview->Lists&Array

28 篇文章

订阅专栏

本文介绍两种不同的方法来删除有序数组中的重复元素，并确保每个元素最多只出现一次或两次。第一种方法适用于元素只能出现一次的情况，而第二种方法允许元素最多出现两次。这两种方法都避免了使用额外的空间，且具有较低的时间复杂度。

Remove Duplicates from Sorted Array I

Given a sorted array, remove the duplicates in place such that each element appear only once and return the new length.

Do not allocate extra space for another array, you must do this in place with constant memory.

For example,
Given input array A = [1,1,2],

Your function should return length = 2, and A is now [1,2].

错误解答：

public class Solution {
    public int removeDuplicates(int[] A) {
        int length = A.length;
        if(length==0 || length==1) return length;
      
        for(int i=1; i<length;i++){
            if(A[i-1]==A[i]){
                for(int j=i; j<length-1; j++) A[j]=A[j+1];
                //A[length-1]='\0';
            
                length--;
                i--;
            }
        }
        return length;
        
    }
}

虽然答案正确，但是O(n)为n的平方，time limit exceeded。

Solution:

public class Solution {
    public int removeDuplicates(int[] A) {
        int length = A.length;
        if(length==0 || length==1) return length;
        int next=1;
        length=1;
        for(int i=1; i<A.length;i++){
            if(A[i-1]!=A[i]){
                A[next] = A[i];
                next++;
                length++;
            }
        }
        return length;
    }
}

这种方法只需遍历一次，分别把只出现了一次的数填到A[next]里，时间复杂度为O(n)。

Remove Duplicates from Sorted Array II

Follow up for "Remove Duplicates":
What if duplicates are allowed at most twice?

For example,
Given sorted array A = [1,1,1,2,2,3],

Your function should return length = 5, and A is now [1,1,2,2,3].

Solution1：

public class Solution {
    public int removeDuplicates(int[] A) {
        int length = A.length;
        if(length<3) return length;
        int times =0; 
        for(int i=1; i< length;i++){
            if(A[i-1]==A[i]) times++;
            else{ times=0;}
            if(times>=2){
                for(int j=i; j<length-1; j++){
                    A[j] = A[j+1];
                }
                //A[length-1]= '\0';
                times--;
                i--;
                length--;
            }
       
        }
        return length;
        
    }
}

这种方法一发现重复两次以上的元素就删掉该元素，将后面元素整体往前移。虽然通过了，但时间复杂度为n的平方，不是很好的方法。

Solution2:

public class Solution {
    public int removeDuplicates(int[] A) {
        int length = A.length;
        if(length<3) return length;
        int times=0, next=1;
        length=1;
        for(int i=1; i< A.length;i++){
            if(A[i-1]==A[i]) times++;
            else{ times=0;}
            if(times<2){
                A[next]=A[i];
                next++;
                length++;
            }
        }
        return length;
    }
}

这种方法只需遍历数组一次，时间复杂度为O(n)。用times记录一个元素重复次数，用next记录移动完的下标，用length记录移动完的长度。如果times<2，就将A[i]的值移动到A[next]，否则什么也不做，只移动下标i。

Note：

看其他同学的答案，有些将新生成的数组尾设为'\0'，如代码中注释语句所示。其实在java中数组没有结束符，即使将数组尾设为'\0'，数组也不会结束，而是将其中元素值设为0。这道题中改动后的数组主要由数组中元素和返回的int值，也就是数组的新长度决定。