LeetCode OJ:Remove Element

Remove Element

 

Given an array and a value, remove all instances of that value in place and return the new length.

The order of elements can be changed. It doesn't matter what you leave beyond the new length.

class Solution {
public:
    int removeElement(int A[], int n, int elem) {
        int count=0;
        for(int i = 0; i < n; i++)
            if (elem != A[i])
            {
                A[count++] = A[i];
            }
        return count;
    }
};

answer2

时间复杂度O(n)，空间复杂度O(1)

class Solution {
public:
    int removeElement(int A[], int n, int elem) {
        return distance(A,remove(A,A+n,elem));
    }
};