LintCode 172. 删除元素

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
        if(A.size()==0)
            return 0;
        for(int i=0;i<A.size();)
        {
            if(A[i]==elem)
            {
                A.erase(A.begin()+i);
            }
            else
            {
                i++;
            }
        }
        return A.size();    
    }
};