[LeetCode] Remove Duplicates from Sorted Array II

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]

 

加一个变量记录一下元素出现的次数即可。这题因为是已经排序的数组，所以一个变量即可解
决。如果是没有排序的数组，则需要引入一个 hashmap 来记录出现次数。

 

方法1  用A[i] 和 A[index] 比较，同时，搞一个计数器

class Solution {
public:
    int removeDuplicates(int A[], int n) {
        if(n == 0)
        return 0;
        
        int index = 0;
        int cnt=1;
        for(int i = 1; i<n; i++)
        {
            if(A[index] != A[i])
            {
                index++;
                A[index]=A[i];
                cnt = 1;
            }
            else
            {
                if(cnt ==1)
                {
                    index++;
                    A[index]=A[i];
                    cnt +=1;
                }
            }
                
        }
        return index + 1;
    }
};

方法2  用A[i] 和 A[index-1] 比较，此方法可推广至最多允许k个数的情况

class Solution {
    public:
        int removeDuplicates(int A[], int n) {
            if (n <= 2) return n;
            int index = 2;//此方法可推广至最多允许k个数的情况,修改inde的值即可
            for (int i = 2; i < n; i++){
                if (A[i] != A[index - 2])
                    A[index++] = A[i];
            }
            return index;
        }
};