如何找到n向量的并集?

时间:2015-04-30 12:49:03

标签: c++ vector set-union

我有一个超边界的二维向量以及一个邻接列表。我必须找到hyperEdges[i].size()向量的并集,但我只能找到两个向量的并集。我可以对下面的代码做些什么改进呢? 我想将联合存储到新声明的二维向量connectedEdges

void find_union()
{
    connectedEdges.resize(nEdges+1);
    for(int i = 1; i <= nEdges; i++)
    {
        vector<int>::iterator it;
        connectedEdges[i].resize(nEdges+1);

        for(int j = 1; j < hyperEdges[i].size()-1; j++)
        {
            int p = hyperEdges[i][j-1];
            int q= hyperEdges[i][j];
            it = set_union(adjL[p].begin(), adjL[p].end(),adjL[q].begin(),adjL[q].end(), connectedEdges[i].begin());
        connectedEdges[i].resize(it-connectedEdges[i].begin());
        }
    }    
}

示例: {1,2,4,6,8}

{1,2,3,5,6}

{1,4,7,13,15}

这三组联盟应为{1,2,3,4,5,6,7,8,13,15} 但我的程序返回{1,2,3,4,5,6,8}

2 个答案:

答案 0 :(得分:6)

如果您有很多向量,我建议将所有这些内容插入到单个std::set中,然后将其转储回std::vector

类似的东西:

std::vector<std::vector<int>> src = ...;
std::set<int> all;

for(int i = 0; i < src.size(); i++) {
    all.insert(src[i].begin(), src[i].end());
}

std::vector<int> result(all.begin(), all.end());

答案 1 :(得分:2)

您还可以使用所有相邻向量运行std::set_union(确保它们已排序),按步骤构建结果,如下例所示:

#include <iostream>
#include <vector>
#include <algorithm>
#include <iterator>

int main()
{
    std::vector<std::vector<int>> v{{1, 1, 2}, {1, 1, 1}, {1, 5, 6, 7}};       
    std::vector<int> result, tmp;
    for(auto&& elem: v)
    {
        std::set_union(elem.begin(), elem.end(), 
                       result.begin(), result.end(),
                       std::back_inserter(tmp));
        result = std::move(tmp); 
        tmp.clear(); // see http://stackoverflow.com/q/9168823/3093378
    }
    // in case you want to remove duplicates
    result.erase(std::unique(result.begin(), result.end()), result.end());
    for(auto&& elem: result)
        std::cout << elem << " "; // 1 2 5 6 7
}