如何根据另一个向量中的条目从向量中删除元素？

Question

我已经搜索了 stackoverflow 并找到了我的问题的部分答案： How to remove elements from a vector based on a condition in another vector? 在我的例子中，我有两个向量，一个包含整数，一个包含共享指针 (std::shared_ptr)，我无法计算出指针算法以使代码正常工作。

class someAbstractClass
{     
};

class someClass : public someAbstractClass
{
};    

std::vector<int> condition;
std::vector<std::shared_ptr< someAbstractClass >> container;

condition.push_back(0);
condition.push_back(1);
condition.push_back(0);

container.push_back( std::make_shared< someClass > );
container.push_back( std::make_shared< someClass > );
container.push_back( std::make_shared< someClass > );

std::erase( std::remove_if( container.begin(), conatiner.end(),
           [&] ( const std::shared_ptr< someAbstractClass > &s ) 
           { 
               return condition[ &s - &(*container.begin()) ] == 1;
           } ), container.end() );

有没有办法让它工作？

Answer 1

您正在将shared_ptr 对象相互比较。您需要比较这些对象的地址。并且 lambda 需要通过 reference 获取其输入参数，以确保它使用向量中原始对象的地址而不是内存中其他位置的对象的 副本 .

container.erase(
    std::remove_if(
        container.begin(), container.end(),
        [&](const std::shared_ptr<someType> &s) {
            return condition[ &s - &(*container.begin()) ] == 1;
        }
    ),
    container.end()
);

Live Demo

将其与您链接到的the accepted answer 到the question 中的以下sn-p 进行比较：

a.erase(remove_if(begin(a), end(a),
    [b&](const myClass& d) { return b[&d - &*begin(a)].alive(); }),
    end(a));

看到相似之处了吗？

减去连续内存容器中两个有效元素的类型地址，例如std::vector，会在两个元素之间产生偏移。并且当较低地址用于容器的第一个元素时，该偏移量实际上与较高地址的索引相同。这就是本示例中的指针算法如何能够在condition 中获取 lambda 参数的索引（前提是condition 至少具有与container 一样多的元素）。