Leetcode_Remove Duplicates from Sorted Array II

来源:互联网 发布:数据库前置库 编辑:程序博客网 时间:2024/05/18 03:23

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

class Solution {public:    int removeDuplicates(int A[], int n) {        if(n<2)        return n;        int start=0,cur=1;        int cnt=0;        while(cur<=n-1){            if(A[start]==A[cur]){                cnt++;                if(cnt>1){                    cur++;                }                else{                    A[++start]=A[cur++];                }            }            else{                cnt=0;                 A[++start]=A[cur++];            }        }        return start+1;    }};


0 0
原创粉丝点击