vector中高效删除元素

最新推荐文章于 2023-12-21 09:47:05 发布

weixin_30662109

最新推荐文章于 2023-12-21 09:47:05 发布

阅读量932

点赞数

CC 4.0 BY-SA版权

文章标签： rust

原文链接：http://www.cnblogs.com/uarekoalaiamtree/p/10897150.html

博客围绕Rust中删除元素展开，介绍了三种方法。一是遍历用erase删除，删除后迭代器会变化，大数据量时效率低；二是两次访问用remove_copy_if和remove_if；三是一次访问用stable_partition。还给出三种方法的效率之比为118 : 1.4 : 1。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1 源码

#include <iostream>
#include <algorithm>
#include <vector>
#include <iterator>
#include <ctime>
#include "stdlib.h"

using namespace std;

// 判断成绩是否及格
bool pgrade(double x)
{
    if(x>=60)
        return true;
    else
        return false;
}

// 判断成绩是否不及格
bool fgrade(double x)
{
    return !pgrade(x);
}

int main()
{
    vector<double> grades_tmp, grades, fail, pass;
    vector<double>::iterator iter;
    clock_t time_start, time_end;
    
    // 在0到100中随机生成75000个成绩
    for (int i=0; i<75000; i++)
        grades_tmp.push_back(rand()%100);
    
    // 循环所有元素，取出不及格成绩后erase
    grades.assign(grades_tmp.begin(), grades_tmp.end());
    time_start = clock();
    iter = grades.begin();
    while (iter < grades.end())
    {
        if(fgrade(*iter))
        {
            fail.push_back(*iter);
            iter = grades.erase(iter);
        }
        else
            iter++;
    }
    pass.assign(grades.begin(), grades.end());
    time_end = clock();
    cout << "erase time is: " << double(time_end-time_start) << endl;
    
    // 两次访问，取出及格和不及格成绩（使用remove_copy_if和remove_if）
    grades.clear();
    fail.clear();
    pass.clear();
    grades.assign(grades_tmp.begin(), grades_tmp.end());
    time_start = clock();
    remove_copy_if(grades.begin(), grades.end(), back_inserter(fail), pgrade);
    grades.erase(remove_if(grades.begin(), grades.end(), fgrade), grades.end());
    pass.assign(grades.begin(), grades.end());
    time_end = clock();
    cout << "twice visit: " << double(time_end-time_start) << endl;
    
    //一次访问，取出及格和不及格成绩（使用stable_partition）
    grades.clear();
    fail.clear();
    pass.clear();
    grades.assign(grades_tmp.begin(), grades_tmp.end());
    time_start = clock();
    iter = stable_partition(grades.begin(), grades.end(), pgrade);
    fail.assign(iter,grades.end());
    grades.erase(iter,grades.end());
    pass.assign(grades.begin(), grades.end());
    time_end = clock();
    cout << "single visit: " << double(time_end-time_start) << endl;
}

2 代码解释

2.1 遍历所有元素，使用erase删除元素

在vector中删除元素时，指向被删除元素和它后面元素的迭代器都失效了；如果添加一个元素，可能导致所有内容重新分配，所有迭代器均失效。因此在循环中使用erase操作时，要特别注意。不过erase删除元素后会返回一个迭代器指向删除元素的下一个元素。
如果vector中的数据量比较大，利用erase删除元素，效率特别低。

2.2 两次访问grades，使用remove_copy_if和remove_if删除元素

对remove_copy_if的理解：某些记录if满足某条件，则remove，copy剩下的到指定位置。当然这里的remove不是真的从vector中删除。
对remove_if的理解：某条记录if满足某条件，则remove，把紧接着、未移动过的不“删除”元素重写在该位置。以remove_if(grades.begin(), grades.end(), fgrade)为例，过程和结果见示意图。