Question

我想从文件流中读取N个字节的数据，并将它们附加到向量中。所以我们说我们有一个

basic_ifstream<uint8_t> myFileStream;

和

vector<uint8_t> myBuffer;

目前我正在做这样的事情：

myBuffer.reserve(N);
for (int i=0; i<N; ++i)
{
    uint8_t tmpByte;
    myFileStream.read(&tmpByte, 1);
    myBuffer.push_back(tmpByte);
}

但这非常慢。

现在我尝试让myFileStream.read将数据直接复制到矢量中。由于向量将其元素存储在连续的存储位置，我认为这样的事情应该是可能的：

uint8_t* ptr = &myBuffer.back(); // there is already some elements in the buffer (I know)
ptr++; // first element after existing data
myBuffer.resize(myBuffer.size() + N);
myFileStream.read(ptr, N);

但是有了这个，我得到一个运行时错误（堆的损坏）。这个解决方案有什么问题？或者还有更好的方法吗？

Answer 1

您的问题是resize可能需要重新分配整个向量，从而使之前的ptr无效。您只需在resize之后使用指针。

std::size_t oldSize = myBuffer.size();
// resize first
myBuffer.resize(oldSize + N);
uint8_t* ptr = &myBuffer[oldSize]; // already first element after existing data
myFileStream.read(ptr, N);

请注意，作为奖励，即使原始向量为空，此实现仍然有效（当然，对于N != 0）。

从文件中读取N个字节并将它们附加到std :: vector

1 个答案: