172. 删除元素

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

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

样例

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

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

 

没什么好说的，用iterator直接过

int removeElement(vector<int> &A, int elem) {
        // write your code here
        vector<int>::iterator it=A.begin();
        while(it!=A.end()){
            if(*it==elem){
                it=A.erase(it);
            }
            else{
                it++;
            }
        }
        return A.size();
    }

如果不考虑语言特性的话，用覆盖大法就可以了

int removeElement(vector<int> &A, int elem) {
        // write your code here
        int num=0;
        for (int i = 0; i<A.size(); i++) {  
            if (A[i] != elem) {  
                A[num++] = A[i];  
            }  
        }  
        return num;
}