我有一个超边界的二维向量以及一个邻接列表。我必须找到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}
答案 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
}