LintCode 题目:删除元素
程序员文章站
2022-03-24 21:45:31
...
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;
即可: