C++ 连接许多 std::vectors 并删除重复项

Question

我正在从外部 API (std::vector) 接收一些整数。

API 通常需要多次调用，因此我需要将连续 API 调用的所有整数累加到一个局部 vector 中。最后，数组的每个元素都必须是唯一的(不需要排序)。

我的代码如下(使用 getNextVector 来“模拟”数据并模拟 API 调用)。

该代码有效，但我希望此操作具有最高性能。我的方法正确吗？

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

std::vector<int> getNextVector(int i) {
    if ( i == 0 ) {
        std::vector<int> v = { 1,2,3 };
        return v; 
    } else if ( i == 1 ) {
        std::vector<int> v = { 3,4,5 };
        return v; 
    } else if ( i == 2 ) {
        std::vector<int> v = { 5,6,7 };
        return v; 
    } else if ( i == 3 ) {
        std::vector<int> v = { 7,8,9 };
        return v; 
    }
}

int count() { return 4; } //we have four vectors

int main(int argc, char** argv) {

    std::vector<int> dest;
    dest.reserve(20); // we can find this, for simplicity hardcode...

    for( int i = 0; i < count(); i++ ) {
        std::vector<int> src = getNextVector(i);
        dest.insert(
            dest.end(),
            std::make_move_iterator(src.begin()),
            std::make_move_iterator(src.end())
        );
    }

    std::sort(dest.begin(), dest.end());
    dest.erase(unique(dest.begin(), dest.end()), dest.end());
/*
    std::copy(
      dest.begin(),
      dest.end(),
      std::ostream_iterator<int>(std::cout, "\n")
    );
*/
    return 0;
}

Answer 1

我认为您可以将 vector 的元素存储在一个集合中。如果不需要排序，您可以使用 unordered_set。只需执行以下操作 -

std::unordered_set<int> integers;

for (int i = 0; i < count; i++) {
    std::vector<int> src = getNextVector(i);

    for (int j = 0; j < src.size(); j++) {
        integers.insert(src[i]);
    }
}

或者按照@StoryTeller 的建议，您可以使用适当的函数代替循环。例如 -

std::unordered_set<int> integers;

for (int i = 0; i < count; i++) {
    std::vector<int> src = getNextVector(i);
    integers.insert(src.begin(), src.end());
}