如果缓冲区大小小于文件大小,如何合并两个文件?

时间:2015-04-01 19:43:10

标签: c++ file stl merge mergesort

如果缓冲区大小小于文件大小,如何合并两个文件? 例如,我有两个带有排序整数的文件

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

1 个答案:

答案 0 :(得分:0)

正如评论中所提到的,每个文件只需要1个数字来实现合并,但是你可能需要编写自己的合并逻辑而不是使用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"
        }
    }