leetcode || 80、Remove Duplicates from Sorted Array II

problem：

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]

.

Hide Tags
Array Two
Pointers

题意：对数组进行去重操作，同一元素最多出现2次，返回新数组大小，同时更新数组。

thinking：

（1）先对数组排序

（2）从左往右遍历数组，出现相同元素开始计数，当重复出现次数超过2次时，用后面的元素覆盖多出的元素（数组往前移动）。

code：

class Solution {
public:
int removeDuplicates(int A[], int n) {
if(n<3)
return n;
int i=0;
int j=0;
sort(A,A+n);
while(j<n)
{
while(j<n&&A[j]==A[++j]); //J定位到最后一个相同元素的下一个位置
if(j-i>2)
{
int num=j-i-2;  //多余的元素的个数
for(int k=0;k<n-j;k++) //往左移动覆盖多余的元素
A[i+2+k]=A[j+k];
j=i+2;    //更新j
n-=num;  //数组减小num
}
i=j;  //更新i
}
return n;

}
};