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

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将数据直接复制到向量中。由于vector将其元素存储在一个连续的存储位置，所以我认为应该可以这样做:

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)，此实现也可以工作。