[英]C++ best way to split vector into n vector
我在這個向量中有一個std::vector<std::string>
我來自txt文件的push_back
字符串,如下所示:
std::string line;
std::vector<std::string> path;
while(getline(fichier, line))
{
path.push_back(line);
}
我想將路徑vector
分成另一個10行的vector
,例如。 因此,如果我的vector
大小是25,我想要另外2個10元素的vector
和5個元素的vector
。
最好的方法是什么?
最好的是意見問題,但你可以做類似以下的事情(使用bunch_size
為10
):
for(size_t i = 0; i < strings.size(); i += bunch_size) {
auto last = std::min(strings.size(), i + bunch_size);
bunches.emplace_back(strings.begin() + i, strings.begin() + last);
}
如果你的字符串很大並且你想避免復制,你可以使用移動版本:
for(size_t i = 0; i < strings.size(); i += bunch_size) {
auto last = std::min(strings.size(), i + bunch_size);
auto index = i / bunch_size;
auto& vec = bunches[index];
vec.reserve(last - i);
move(strings.begin() + i, strings.begin() + last, back_inserter(vec));
}
我提出了一些相當普遍的東西(它適用於不同的容器和不同類型,在這種情況下復雜性會發生變化):
#include <algorithm>
#include <iterator>
#include <vector>
template<typename Vector>
auto split_vector(const Vector& v, unsigned number_lines) {
using Iterator = typename Vector::const_iterator;
std::vector<Vector> rtn;
Iterator it = v.cbegin();
const Iterator end = v.cend();
while (it != end) {
Vector v;
std::back_insert_iterator<Vector> inserter(v);
const auto num_to_copy = std::min(static_cast<unsigned>(
std::distance(it, end)), number_lines);
std::copy(it, it + num_to_copy, inserter);
rtn.push_back(std::move(v));
std::advance(it, num_to_copy);
}
return rtn;
}
您可以指定要拆分的行數:
例如:
int main(int argc, char *argv[]) {
std::vector<std::string> input_vector = {"First", "Second", "Third"};
auto vs = split_vector(input_vector, 2);
return 0;
}
它將產生兩個向量: {"First", "Second"}
和{"Third"}
。
您可以在讀取文件時使用流迭代器來完成工作:
using packet_t = Packet<5>;
using filler_t = std::istream_iterator<packet_t>;
std::vector<packet_t> packets{
filler_t(stream),
filler_t()
};
用Packet
聲明所需的結構operator>>
:
template<size_t size>
struct Packet
{
std::vector<std::string> lines;
friend std::istream& operator>>(std::istream& is, Packet& packet)
{
packet.lines.clear();
std::string line;
for(size_t i = 0; i < size && std::getline(is, line); ++i)
{
packet.lines.push_back(line);
}
if(packet.lines.size() > 0)
{
is.clear();
}
return is;
}
};
請注意,當數據包對於最后一行不為空時,將清除該流。
完整代碼:
#include <iostream>
#include <iterator>
#include <sstream>
#include <vector>
template<size_t size>
struct Packet
{
std::vector<std::string> lines;
friend std::istream& operator>>(std::istream& is, Packet& packet)
{
packet.lines.clear();
std::string line;
for(size_t i = 0; i < size && std::getline(is, line); ++i)
{
packet.lines.push_back(line);
}
if(packet.lines.size() > 0)
{
is.clear();
}
return is;
}
};
int main()
{
std::istringstream stream("1\n2\n3\n4\n5\n6\n7\n");
using packet_t = Packet<5>;
using filler_t = std::istream_iterator<packet_t>;
std::vector<packet_t> packets{
filler_t(stream),
filler_t()
};
for(auto& packet : packets)
{
for(auto& line : packet.lines)
{
std::cout << line << " ";
}
std::cout << std::endl;
}
}
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.