题目:
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].
思路
两个指针,i用于遍历A数组,index指向B数组中下一个可能的合法位置,初始设置为index=2。
比较A[i]和B[index-2],若相等,说明i指向的元素不合法,此时不应该把A[i]加入到B中,i++;
若不想等,说明i指向的元素合法,应该设置B[index] = A[i],index++,指向下一个可能合法的位置。
java代码
public int removeDuplicates2(int A[]) {
int n = A.length;
int[] B = new int[n];/
if (n < 3) // 最多含有2个元素,一定满足题意
return n;
B[0] = A[0];
B[1] = A[1];
int index = 2; // 可能不合法的位置
for (int i = 2; i < n; i++) {
if (A[i] != B[index - 2])
B[index++] = A[i];
}
for(int i=0;i<n;i++)
A[i] = B[i];
return index;
}