LeetCode - Remove Duplicates from Sorted Array

Remove Duplicates from Sorted Array

2013.12.7 03:29

Given a sorted array, remove the duplicates in place such that each element appear only once and return the new length.

Do not allocate extra space for another array, you must do this in place with constant memory.

For example,
Given input array A = [1,1,2],

Your function should return length = 2, and A is now [1,2].

Solution:

　　The array is sorted, so duplicates must be in a consecutive group. Here I used two pointers, one to scan the array, the other to record the unique value. Everytime a unique value is found, record it. The rest of the duplicate values are discarded and overwritten.

　　Time complexity is O(n), space complexity is O(1).

Accepted code:

//#define __MAIN__
#include <cstdio>
#include <cstdlib>
using namespace std;

class Solution {
public:
    int removeDuplicates(int A[], int n) {
        // Note: The Solution object is instantiated only once and is reused by each test case.
        int i, j, k;

        i = 0;
        k = 0;
        while(i < n){
            j = i + 1;
            while(j < n && A[i] == A[j]){
                ++j;
            }
            A[k] = A[i];
            ++k;
            i = j;
        }

        return k;
    }
};

#ifdef __MAIN__
int main()
{
    Solution sol;
    int A[] = {1};
    const int n = sizeof(A) / sizeof(int);
    int len;

    len = sol.removeDuplicates(A, n);
    printf("%d\n", len);
    for(int i = 0; i < len; ++i){
        printf("%d ", A[i]);
    }
    printf("\n");

    return 0;
}
#endif