如何正确地删除vector中符合条件的某元素

如何正确地删除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;
    }
}
  • 2
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值