如何在C ++循环中从向量中删除元素

时间:2018-01-18 19:49:58

标签: c++ algorithm vector stl erase

我有2个双倍向量:tP。它们的大小为m

我想检查条件:向量|t[i]-t[i+1]| < dT的{​​{1}}和向量t的{​​{1}}。

如果条件正确,我应该删除|P[i]-P[i+1]| < dP元素(或P元素)。

我的代码:

t[i+1]

当我使用P[i+1]按索引删除时,这是正确的方法吗?

1 个答案:

答案 0 :(得分:1)

对于这样的任务,最好将标准算法std::unique与谓词一起使用,然后将成员函数erase应用于返回的迭代器。

至于你的代码,它是无效的。删除元素时不应增加索引。

这是一个演示程序,显示如何使用算法std::unqiue

#include <iostream>
#include <vector>
#include <algorithm>
#include <cstdlib>

int main() 
{
    std::vector<double> v = { 1, 1.5, 3, 4.5, 5 };
    const double delta = 1.0;

    for ( const auto &x : v ) std::cout << x << ' ';
    std::cout << std::endl;

    v.erase( 
        std::unique( v.begin(), v.end(), 
                     [&]( const auto &x, const auto &y ) 
                     { 
                        return ::abs( x - y ) < delta; 
                     } ),
        v.end() );


    for ( const auto &x : v ) std::cout << x << ' ';
    std::cout << std::endl;

    return 0;
}

它的输出是

1 1.5 3 4.5 5 
1 3 4.5