LeetCode之RemoveDuplicates扩展

题目：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]

分析：由于不能另外定义一个数组，根据昨天那个简单高效的代码，可以在其基础上发散。

代码：

//要求已排序的数组中，最多只允许出现2次相同的数
class Solution {
public:
	int removeDuplicates(int A[], int n) {
		int index = 2;
		for (int i=0;i<n;i++)
		{
			if (A[i] != A[index-2])
			{
				A[index++] = A[i];
			}
		}
		return index;
	}

};

思考：从这个代码可以看出，如果要求最多只能重复N次，那么就只需要将2改为N,即可。

下面是我第一思路的实现方法：

思路：既然已经排好序，那么我只要将多于2次的数进行删除，然后移动一位即可。

代码：

class Solution {
public:
	int removeDuplicates(int A[], int n) {
		int len = n;
		int i,j,k;
		for (i = 0;i<len;i++)
		{
			j=i+2;
			while(j<len && A[i] == A[j]){
				for (k=j;k<len-1;k++)
				{
					A[k] = A[k+1];
				}
				A[len] = 0;
				len--;
			}
		}
		A[len] =0;
		return len;
	}
};

做了很多测试，虽然没问题，我只想说，代码写的晦涩难懂，质量低下。不过这也是我刷LeetCode的原因。