如何正确地删除vector中符合条件的某元素
比如,有一个vector nums = {1, 2, 2, 2, 2, 3, 5},要求删除nums中所有值为2的元素。
一个符合直觉的错误实现:
for (vector<int>::iterator it = nums.begin(); it != nums.end(); it++) {
if (*it == 2) {
nums.erase(it);
}
}
这段代码循环遍历nums中的每个元素,判断是否为2,是的话则erase掉。看起来好像没什么问题,但是实际上已经造成了bug。这段代码的执行完成后,nums存储的元素是{1, 2, 2, 3, 5},值为2的元素并没有被全部清除掉,这个结果大家可以自己试验一下。
为什么会出现这个结果呢?原因就是迭代器失效:在第一个2被erase掉的时候,it迭代器已经失效了,用它来继续遍历vector就会漏掉被删除元素后面的第一个元素,导致2没有被完全清除。
第一种正确实现:让it指向下一个元素
erase函数的返回值是指向当前被删除元素的下一个元素的迭代器。那么我们把这个返回值赋值给it继续遍历
for (vector<int>::iterator it = nums.begin(); it != nums.end();) {
if (*it == 2) {
it = nums.erase(it);
} else {
++it;
}
}
第二种正确实现:erase-remove
int main(){
vector<int> nums = {1, 2, 2, 2, 2, 3, 5};
nums.erase(remove(nums.begin(),nums.end(), 2), nums.end());
for (int i = 0; i < nums.size(); ++i) {
cout<<nums[i]<<" ";
}
}
erase用来删除一段区间的元素
iterator erase( const_iterator first, const_iterator last );
remove移除但不删除
template< class ForwardIt, class T >
ForwardIt remove( ForwardIt first, ForwardIt last, const T& value );
remove移除[first, last)之中所有与value相等的元素,它并不真正从容器中删除那些元素(换句话说容器的大小不变),而是将每一个不与value相等(也就是我们不打算移除)的元素轮番赋值给first之后的空间。
例如:{0,1,0,2,0,3,0,4},用remove移除值为0的元素,执行结果为{1,2,3,4,0,3,0,4}
template<class ForwardIterator, class T>
ForwardIterator remove(ForwardIterator first, ForwardIterator last, const T &value) {
first = find(first, last, value);
ForwardIterator next = first;
if (first == last) {
return first;
} else return remove_copy(++next, last, first, value);
}
remove_copy
template< class ForwardIt, class T >
ForwardIt remove( IntputIt first,IntputIt last, OutputIt result, const T& value );
remove移除[first, last)之中所有与value相等的元素,它并不真正从容器中删除那些元素(换句话说容器的大小不变),而是将结果复制到一个以result标示起始位置的容器上,新容器可以和原容器重叠。
template <class InputIt, class OutputIt, class T>
OutputIt remove_copy(InputIt first, InputIt last, OutputIt result, const T& value){
for (;first!=last;++first){
if (*first!=value){
*result = *first;
++result;
}
return result;
}
}