C ++中的矢量交集

时间:2013-10-20 22:33:31

标签: c++ vector intersection

我有这个功能

vector<string> instersection(const vector<string> &v1, const vector<string> &v2);

我有两个字符串向量,我想找到两者中存在的字符串,然后用常见的元素填充第三个向量。

如果我的载体是......

v1 = <"a","b","c">
v2 = <"b","c">

3 个答案:

答案 0 :(得分:41)

尝试std::set_intersection,例如:

#include <algorithm> //std::sort
#include <iostream> //std::cout
#include <string> //std::string
#include <vector> //std::vector

std::vector<std::string> intersection(std::vector<std::string> &v1,
                                      std::vector<std::string> &v2){
    std::vector<std::string> v3;

    std::sort(v1.begin(), v1.end());
    std::sort(v2.begin(), v2.end());

    std::set_intersection(v1.begin(),v1.end(),
                          v2.begin(),v2.end(),
                          back_inserter(v3));
    return v3;
}

int main(){
    std::vector<std::string> v1 {"a","b","c"};
    std::vector<std::string> v2 {"b","c"};

    auto v3 = intersection(v1, v2);

    for(std::string n : v3)
        std::cout << n << ' ';
}

答案 1 :(得分:3)

您只需要对较小的矢量进行排序。然后对较大的向量进行单次传递,并使用二进制搜索在较小的向量中测试其项目的存在。

答案 2 :(得分:0)

与其进行排序,不如考虑用时间来交换内存,方法是从较小的向量中生成一个哈希集,然后遍历较大的向量以检查那些元素,如建议here所示。那将比排序和使用std::set_intersection更快。