如果缓冲区大小小于文件大小,如何合并两个文件?例如,我有两个带有排序整数的文件
1.txt和2.txt
2 1
5 7
6 8
7 9
我必须将它们合并到一个排序的文件中,但从每个文件中读取的数字不能超过两个(这就是任务)。我的记忆中不能同时有4个以上的数字。
这是我的代码
#include <iostream>
#include <fstream>
#include <vector>
#include <iterator>
#include <algorithm>
using namespace std;
const int bufferSize = 2;
bool readSortFile(ifstream &file, vector<int> &data) {
int tmp;
for (int i = 0; (i < bufferSize && file >> tmp); i++)
data.push_back(tmp);
return file.good();
}
int main() {
ifstream file1("1.txt");
ifstream file2("2.txt");
ofstream out;
vector<int> data1, data2;
bool fileGood1, fileGood2;
fileGood1 = true;
fileGood2 = true;
while(fileGood1 || fileGood2) {
if(data1.size() == 0)
fileGood1 = readSortFile(file1, data1);
if (data2.size() == 0)
fileGood2 = readSortFile(file2, data2);
out.open("temp", ios::app);
merge(data1.begin(),
data1.end(),
data2.begin(),
data2.end(),
ostream_iterator<int>(out, "n"));
data1.clear();
data2.clear();
out.close();
}
rename("temp", "result.txt");
file1.close();
file2.close();
return 0;
}
输出为1 2 5 7 6 7 8 9
如注释中所述,您只需要每个文件中的一个数字就可以实现合并,但您可能需要编写自己的合并逻辑,而不是使用std::merge。例如伪代码,您需要添加eof检查(如果1.txt结束,则复制2.txt的其余部分,反之亦然)。
num1 = get number from "1.txt"
num2 = get number from "2.txt"
loop(...){
if(num1 <= num2){
write num1
num1 = get number from "1.txt"
} else {
write num2
num2 = get number from "2.txt"
}
}