LeetCode之Remove Duplicates from Sorted Array II

/*此题可以参考Remove Duplicates from Sorted Array的
方法一，只要加入一种新的情况：当A[k]==A[i]&&A[k]!=A[k-1]时，即可
加入新的元素到结果数组。
方法参考自：https://github.com/soulmachine/leetcode*/
class Solution {
public:
	int removeDuplicates(int A[], int n) {
		if(n <= 2) return n;
		int k(1);
		for(int i = 2; i < n; ++i){
			if((A[k] != A[i]) || (k > 0 && A[k] != A[k-1])){
				A[++k] = A[i];
			}
		}
		return ++k;
	}
};