我正在使用多步操作，并希望合并结果。例如:
std::vector<int> A;
std::vector<int> B;
std::vector<int> AB;
我想让AB有A的内容和B的内容按照这个顺序。做这样的事情最有效的方法是什么?
当前回答
基于Kiril V. Lyadvinsky的回答，我做了一个新的版本。这个片段使用模板和重载。有了它，你可以写vector3 = vector1 + vector2和vector4 += vector3。希望能有所帮助。
template <typename T>
std::vector<T> operator+(const std::vector<T> &A, const std::vector<T> &B)
{
std::vector<T> AB;
AB.reserve(A.size() + B.size()); // preallocate memory
AB.insert(AB.end(), A.begin(), A.end()); // add A;
AB.insert(AB.end(), B.begin(), B.end()); // add B;
return AB;
}
template <typename T>
std::vector<T> &operator+=(std::vector<T> &A, const std::vector<T> &B)
{
A.reserve(A.size() + B.size()); // preallocate memory without erase original data
A.insert(A.end(), B.begin(), B.end()); // add B;
return A; // here A could be named AB
}
其他回答
还有一个没有提到的简单变体:
copy(A.begin(),A.end(),std::back_inserter(AB));
copy(B.begin(),B.end(),std::back_inserter(AB));
并使用合并算法:
#include <algorithm>
#include <vector>
#include <iterator>
#include <iostream>
#include <sstream>
#include <string>
template<template<typename, typename...> class Container, class T>
std::string toString(const Container<T>& v)
{
std::stringstream ss;
std::copy(v.begin(), v.end(), std::ostream_iterator<T>(ss, ""));
return ss.str();
};
int main()
{
std::vector<int> A(10);
std::vector<int> B(5); //zero filled
std::vector<int> AB(15);
std::for_each(A.begin(), A.end(),
[](int& f)->void
{
f = rand() % 100;
});
std::cout << "before merge: " << toString(A) << "\n";
std::cout << "before merge: " << toString(B) << "\n";
merge(B.begin(),B.end(), begin(A), end(A), AB.begin(), [](int&,int&)->bool {});
std::cout << "after merge: " << toString(AB) << "\n";
return 1;
}
这正是成员函数std::vector::insert的作用
std::vector<int> AB = A;
AB.insert(AB.end(), B.begin(), B.end());
如果你的向量排序*，检查set_union从<algorithm>。
set_union(A.begin(), A.end(), B.begin(), B.end(), AB.begin());
链接中有一个更详细的例子。
template<typename T>
vector<T> CombineVectors(vector<T> &&a, vector<T> &&b) {
vector<T> ab = move(a);
ab.insert(
ab.end(), make_move_iterator(b.begin()), make_move_iterator(b.end()));
return ab;
}
template<typename T>
vector<T> CombineVectors(const vector<T> &a, const vector<T> &b) {
vector<T> ab = a;
ab.insert(ab.end(), b.begin(), b.end());
return ab;
}
基于Kiril V. Lyadvinsky的回答，我做了一个新的版本。这个片段使用模板和重载。有了它，你可以写vector3 = vector1 + vector2和vector4 += vector3。希望能有所帮助。
template <typename T>
std::vector<T> operator+(const std::vector<T> &A, const std::vector<T> &B)
{
std::vector<T> AB;
AB.reserve(A.size() + B.size()); // preallocate memory
AB.insert(AB.end(), A.begin(), A.end()); // add A;
AB.insert(AB.end(), B.begin(), B.end()); // add B;
return AB;
}
template <typename T>
std::vector<T> &operator+=(std::vector<T> &A, const std::vector<T> &B)
{
A.reserve(A.size() + B.size()); // preallocate memory without erase original data
A.insert(A.end(), B.begin(), B.end()); // add B;
return A; // here A could be named AB
}
