将文件读入 std::vector<char> 的有效方法?

2021-12-21 00:00:00 vector c++ stl

我想避免不必要的副本.我的目标是:

I'd like to avoid unnecessary copies. I'm aiming for something along the lines of:

std::ifstream testFile( "testfile", "rb" );
std::vector<char> fileContents;
int fileSize = getFileSize( testFile );
fileContents.reserve( fileSize );
testFile.read( &fileContents[0], fileSize );

(这不起作用,因为 reserve 实际上并没有在向量中插入任何东西,所以我无法访问 [0]).

(which doesn't work because reserve doesn't actually insert anything into the vector, so I can't access [0]).

当然,std::vectorfileContents(fileSize) 有效,但初始化所有元素会产生开销(fileSize 可能相当大).resize() 也一样.

Of course, std::vector<char> fileContents(fileSize) works, but there is an overhead of initializing all elements (fileSize can be rather big). Same for resize().

这个问题与开销的重要性无关.相反,我只是想知道是否还有其他方法.

This question is not so much about how important that overhead would be. Rather, I'm just curious to know if there's another way.

推荐答案

规范形式是这样的:

#include<iterator>
// ...

std::ifstream testFile("testfile", std::ios::binary);
std::vector<char> fileContents((std::istreambuf_iterator<char>(testFile)),
                               std::istreambuf_iterator<char>());

如果您担心重新分配,请在向量中保留空间:

If you are worried about reallocations then reserve space in the vector:

#include<iterator>
// ...

std::ifstream testFile("testfile", std::ios::binary);
std::vector<char> fileContents;
fileContents.reserve(fileSize);
fileContents.assign(std::istreambuf_iterator<char>(testFile),
                    std::istreambuf_iterator<char>());

相关文章