Remove Duplicates from Sorted Array II (Java)

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

这道题比之前那道变化在最多允许重复两次。上道题用的方法是，两个指针指向前后两个元素，如果元素不同，则替换flag位的数字，由于flag走的比较慢，所以不存在把数字冲掉而造成错误。而这道题就存在这种现象，设两个指针，中间隔一位。比如1,1,1,2,2,3，两个指针先分别指向第0位和第2位，然后向后走，遇到不等的就记录在flag位中，到第4位的2时，如果保存到flag位就会变成1,1,2,2,2,3，这样比较第2位和第4位的时候就会出错。方法是遇到不同的时候先不替换flag位，而是在下一次替换，这样就保证了下一次先比较，比较完了才替换这一位。

Source

public class Solution {
	public int removeDuplicates(int[] A) {
	    if(A.length <= 2) return A.length;
	    int temp = A[1], flag = 1;
	    for(int i = 2; i < A.length; i++){
	    	if(A[i] != A[i - 2]){	//两步的比较容易出现把原有数字冲掉的可能性，应该先保存要覆盖的新数字，下一次再付给该位置
	    		A[flag] = temp;
	    		flag ++;
	    		temp = A[i];	
	    	}
	    	
	    }
	    A[flag] = temp;
	    return flag + 1;   //下标数+1 因为下标是从0开始
	    
	}
}

Test

    public static void main(String[] args){
    	int[] A = {1,1,1,1,1,2,2};
    	int b = new Solution().removeDuplicates(A);
    	for(int i = 0; i < b ; i++)
    		System.out.println(A[i]);
    }