LeetCode-Remove Duplicates from Sorted Array II （删除重复元素，使重复次数最多N次）-蒲公英云

LeetCode-Remove Duplicates from Sorted Array II （删除重复元素，使重复次数最多N次）

电玩女神 2022-08-07 03:54 139阅读 0赞

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].

可以先看另外一道题，http://blog.csdn.net/my_jobs/article/details/43453177，这个是加强版的。其实仿照上题，改造好的代码如下：

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

只是在if语句中，加了更严谨的条件判断。

此题可以再加强一下，如果重复元素最后允许出现3次呢？5次呢？或者直接n次？怎么解？可以说思路基本一样，只不过需要判断新数组中重复元素的个数，如果超过n就不需要复制了。上代码：

public int removeDuplicates(int[] A, int n) {
        if (A.length < n+1) {
            return A.length;
        }
        int j = 0; 
        int s = 0;
        for (int i = 1; i < A.length; i++) {
            if (A[j] != A[i]) {
                A[++j] = A[i];
                s = j;
            } else if (A[j] == A[i] && j - s < n-1) {
                A[++j] = A[i];
            }
        }
        return j+1;
    }