172. 删除元素

Posted 2020-10-19 三人木君

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

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

样例

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

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

 

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

 1 int removeElement(vector<int> &A, int elem) {
 2         // write your code here
 3         vector<int>::iterator it=A.begin();
 4         while(it!=A.end()){
 5             if(*it==elem){
 6                 it=A.erase(it);
 7             }
 8             else{
 9                 it++;
10             }
11         }
12         return A.size();
13     }

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

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;
}