leetcode[80] Remove Duplicates from Sorted Array II

给定一个排好序的数组，要求里面数字重复的次数不超过2，并且记录在原数组的前头，返回剩余长度。例如给定:

A = [1,1,1,2,2,3]　　返回 5，并且A = [1,1,2,2,3]

思路：

用till记录满足条件的下一个位置，以便下一次填入

用repeat记录重复的次数，超过2则不理，否则往till里记录

如果n为0，则返回0，如果非空，那么第一个数肯定是符合的，所以till从1开始，repeat从1开始。

class Solution {
public:
    int removeDuplicates(int A[], int n) {
        if (n == 0) return 0;
        int till = 1, repeat = 1;
        for (int i = 1; i < n; i++)
        {
            if (A[i] == A[i-1])
                repeat++;
            else
                repeat = 1;
            if (repeat <= 2)
            {
                A[till++] = A[i];
            }
        }
        return till;
    }
};

 2015/03/29：

就是设置两个标记，下一个数要不等于前两个数才能加入，其实还是上面的方法好，它可以拓展到多个重复的情况，下面这个Python的就不能了：

class Solution:
    # @param A a list of integers
    # @return an integer
    def removeDuplicates(self, A):
        if (len(A) < 3):
            return len(A)
        start1, start2 = 0, 1
        for i in range(2, len(A)):
            if not(A[start1] == A[i] and A[start2] == A[i]):
                start1 = start2;
                start2 += 1
                A[start2] = A[i]
        return start2 + 1