【代码优化】如何高效地删除vector中的元素普通遍历删除 VS std::stable_partition

奋起的熊猫

于 2024-03-21 10:55:09 发布

阅读量235

点赞数 1

文章标签：算法

本文链接：https://blog.csdn.net/gaochubusheng/article/details/136901885

版权

本文对比了C++中std::stable_partition删除vector中特定元素的方法与普通遍历删除的性能，通过测试数据展示了stable_partition在效率上的优势。

摘要由CSDN通过智能技术生成

场景

删除vector中不符合条件的元素

方式

std::stable_partition
实现原理：按照某种过滤规则，将vector中的元素分成两部分，前半部分是符合条件的，后半部分是不符合条件的。
具体实现：

void test1(int n)
{
    vector<int> vec;
    constructVec(vec, n);

    clock_t start, end;
    start = clock();
    auto itr = std::stable_partition(vec.begin(), vec.end(), [](const int val){
        return val == 0;
    });
    size_t delCount = std::distance(itr, vec.end());
    if (delCount > 0) {
        vec.erase(itr, vec.end());
    }
    end = clock();
    cout << "useTime1: " << end - start << endl;
}

普通遍历删除

void test2(int n)
{
    vector<int> vec;
    constructVec(vec, n);
    clock_t start, end;
    start = clock();
    for (auto itr = vec.begin(); itr != vec.end(); ) {
        if (*itr != 0) {
            itr = vec.erase(itr);
        }else{
            itr++;
        }
    }
    end = clock();
    cout << "useTime2: " << end - start << endl;
}