Remove Element

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