[LintCode]172.删除元素

给定一个数组和一个值，在原地删除与值相同的数字，返回新数组的长度。

元素的顺序可以改变，并且对新的数组不会有影响。

样例

给出一个数组 [0,4,4,0,0,2,4,4]，和值 4

返回 4 并且4个元素的新数组为[0,0,0,2]

思路：类似上一题100

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