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