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