[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].

Hide Tags

 Array Two Pointers

 

---

    这个很容易判断了，设两个index，一个是遍历的，一个是指向返回的最后位置，因为已经排序了，所以判断索引的值与最后位置及前一个的值是否相等，相等继续遍历，不相等更新索引。

#include <iostream>
using namespace std;

class Solution {
public:
    int removeDuplicates(int A[], int n) {
        if(n<3) return n;
//        cout<<n<<endl;
        int retidx= 1,curidx =2;
        for(;curidx<n;curidx++){
//            cout<<retidx<<" "<<curidx<<endl;
            if(A[retidx]==A[curidx]&&A[retidx-1]==A[curidx])
                continue;
            A[++retidx] = A[curidx];
        }
        return retidx+1;
    }
};

int main()
{
    int a[]={};
    Solution sol;
    int ret = sol.removeDuplicates(a,sizeof(a)/sizeof(int));
    for(int i=0;i<ret;i++)
        cout<<a[i]<<" ";
    cout<<endl;
    return 0;
}