Question

我对这段代码有一些问题，代码正在编译，但是当我尝试测试向量的内容时，测试失败了。这是代码：

using std::vector;

// returns a different vector, containing all elements in v1 and all elements in v2                          (elements who are either in v1 or v2) but no duplicates.

template <typename T> vector<T> set_union(const vector<T>& v1, const vector<T>&v2)
{
    vector<T> v(20);
    typename vector<T>::iterator it;

    it = set_union (v1.begin(), v1.end(), v2.begin(), v2.end(), v.begin());
    return v;
}

这是我正在进行的测试：

TEST_F(MyTest,set_union) {
        vector<int> v1{1,3,2};
        vector<int> v2{1,4};
        vector<int> v=set_union(v1,v2);
        ASSERT_EQ(0,count(v,9));
        ASSERT_EQ(1,count(v,1));
        ASSERT_EQ(1,count(v,2));
        ASSERT_EQ(1,count(v,3));
        ASSERT_EQ(1,count(v,4));

}

当我运行这些测试时，第一个测试通过，但第二个测试返回向量中数字1的0个实例，其中答案应为1个实例。

Answer 1

问题是std::set_union要求对输入数据进行排序，而v1则不是。

编辑：正如评论中所述，你不应该预先调整你的矢量大小，因为它会结束一堆0，除非你在联合结果中只有20个项目。相反，这样的事情是什么（我更改了名称以使其更具描述性，并且在调用中使用忍者编辑来保留基于注释的最小大小限制）：

template <typename T>
std::vector<T> vector_union(const std::vector<T>& v1, const std::vector<T>& v2)
{
    vector<T> v;
    v.reserve(std::max(v1.size(), v2.size());

    set_union (v1.begin(), v1.end(), v2.begin(), v2.end(), std::back_inserter(v));

    return v;
}

如果需要，你甚至可以在调用set_union之前对这些向量进行排序，但这会导致预先排序的输入不需要工作。

c ++ set_union迭代器

1 个答案: