LeetCode Remove Duplicates from Sorted Array II

LeetCode Remove Duplicates from Sorted Array II

class Solution {

public:

    int removeDuplicates(vector<int>& nums) {

        int i,k=1;

        int n=nums.size();

        if(n==0)

         return 0;

        if(n==1)

         return 1;

        int prev=nums[0];

        int cnt=0;

        for(i=1;i<n;i++){

            if(nums[i]==prev && cnt==0){

              nums[k]=nums[i];

              cnt=1;

              k++;

            }

            else if(nums[i]!=prev){

                 cnt=0;

                 nums[k]=nums[i];

                 prev=nums[i];

                 k++;

             }

        }

        return k;

    }

};