首页 > 解决方案 > 插入要设置的向量项

问题描述

我想在 C++ 中将字符串项插入vector到字符串中。set如果向量的项目包含'+',则应将其添加到一个集合中,否则应将其添加到另一个集合中。我可以通过一个一个地迭代向量来做到这一点。但是我可以通过 ' callback' 的 ' ' 的 ' ' 的功能来insert做到这一点set吗?

struct compare {
    bool operator() (const std::string& str) const{
        if(str.find("+") != std::string::npos) {
            return false;
        return true;
    }
};
std::vector strlist = {"apple","+apple","banana","orange","+graphes"};
std::set<std::string,compare> set1,set2;
set1.insert(strlist.begin(),strlist.end(),compare);

我想创建一个set1包含项目的列表,

apple,
banana,
orange

以及set2以下项目,

+apple
+graphes

标签: c++vectorstlset

解决方案


您可以std::partition_copy按如下方式使用:

std::partition_copy(strlist.begin(), strlist.end(),
                    std::inserter(set1, set1.end()),
                    std::inserter(set2, set2.end()),
                    [](const std::string& str) { return str.find("+") != std::string::npos; }))

推荐阅读