[leetcode]_Remove Duplicates from Sorted Array II

题目：一个有序数组，要求保证数组中的每个元素不能超过2个。

　　  输入：A = [1,1,1,2,2,3]  输出：length = 5, and A is now [1,1,2,2,3]

 

思路：双指针 。有些绕，不过理清了后，思路还是很直接明朗的。

1、两个指针：p和help。初始化时同时指向数组头。变量cur记录当前元素值，count记录当前元素值出现的次数。

2、当 A[p] == cur && count < 2 或者 A[p] != cur 的时候，A[help] = A[p], help++ , p++。

3、当 A[p] == cur && count >= 2时，p++,help不动。

4、直到p指向尾部，此时help的值即为去重后的数组长度。

 

代码：

public int removeDuplicates(int[] A) {
       if(A.length == 0) return 0;
       
       int cur = A[0] , count = 1;
       int help = 1;
       for(int p = 1 ; p < A.length ; p++){
           if(cur == A[p]){
               if(count < 2){
                   count++;
                   A[help++] = A[p];
               }
           }else{
               count = 1;
               cur = A[p];
               A[help++] = A[p];
           }
       }
       return help;
}