172. Remove Element

Code(Language:C++) (Judger:ip-172-31-18-231)

class Solution {
public:
    /*
     * @param A: A list of integers
     * @param elem: An integer
     * @return: The new length after remove
     */
    int removeElement(vector<int> &A, int elem) {
        // write your code here
        int start = 0;
        for(int i = 0; i < A.size(); i++){
            if(A[i] != elem){
                A[start++] = A[i]; 
            }
        }
        return start; 
    }
};