首页 > 代码库 > leetcode[80] Remove Duplicates from Sorted Array II

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;    }};

 

leetcode[80] Remove Duplicates from Sorted Array II