LintCode 题目：删除元素

URL:https://www.lintcode.com/problem/remove-element/description

描述

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

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

样例

Example 1:
	Input: [], value = 0
	Output: 0


Example 2:
	Input:  [0,4,4,0,0,2,4,4], value = 4
	Output: 4
	
	Explanation: 
	the array after remove is [0,0,0,2]

 

1.失败

在代码段中添加：

for(vector<int>::iterator it = A.begin(); it!=A.end(); it++) {
		    if(*it==elem){
                A.erase(it);
            }
	    }
        return A.size();

得到：

删除的不干净，还有剩余，原因未知。

 

2.成功

在代码段中添加：

vector<int> B;
        for(vector<int>::iterator it = A.begin(); it!=A.end(); it++) {
		    if(*it!=elem){
                B.push_back(*it);
            }
	    }
	    A.clear();
	    A=B;

即可：