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

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]

.

可以先看另外一道题，/article/10984864.html，这个是加强版的。其实仿照上题，改造好的代码如下：

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;
}

i和j的含义同上，只不过加了一个s起始位，表示此元素的起始位，用j-s来计算重复元素的个数。