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

可以先看另外一道题，http://blog.youkuaiyun.com/my_jobs/article/details/43453177，这个是加强版的。其实仿照上题，改造好的代码如下：

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来计算重复元素的个数。