Замена строк в текстовых документах больших размеров - C++ - Ответ 5023367

// без зависимости от '\n' в конце файла
 
#include <windows.h>
#include <iostream>
#include <fstream>
#include <string>
 
using namespace std;
 
const long long int N = 100000000000;
 
const char* filename = "1.txt";
 
void createfile() //создание файла для теста
{
    //string str = "AAAAAAAAAA";
    ofstream fout("test.txt");
    for (int i = 0; i < N; ++i)
    {
        fout << i + 1 << endl;
        //fout << str << endl;
    }
    fout.close();
}
 
char* findStr(char *begin, int n, int w, unsigned long &c) // поиск начала n-ой строки  
                                                           // w - это номер строки, от которой начинается поиск
                                                           // c - счётчик байтов
{
    while (w != n)
    {
        if (*begin == '\n') ++w;
        ++begin;
        ++c;
    }
    
    return begin;
}
 
int numberStr(char *pvFileSrc, unsigned long fileSize, int &flag_n) // подсчёт количества строк
{
    int n = 0;           // количество строк
    unsigned long c = 0; // счётчик байтов
    while (c != fileSize)
    {
        if (*pvFileSrc == '\n') ++n;
        ++pvFileSrc;
        ++c;
    }
    --pvFileSrc;
    if (*pvFileSrc == '\n') // если последняя строка заканчивается на '\n'
    {
        flag_n = 1;
        return n;
    }
    return n + 1;
}
 
void transfer(char* begin, char* end, int offset) // перемещение данных между переставляемыми строками
{
    if (offset > 0) // если первая строка меньше второй
    {
        --end;
        while (end >= begin)
        {
            *(end + offset) = *end;
            --end;
        }
    }
    else // offset отрицательный
    {
        while (begin != end)
        {
            *(begin + offset) = *begin;
            ++begin;
        }
    }
}
 
void showStr(int num1, int num2) // вывод двух строк из файла
{
    FILE *fin = fopen(filename, "rb"); // в бинарном быстрее
    
    if (!fin) 
    {
        printf("Error!\n");
        return;
    }
    else
    {
        char str1[500];
        char str2[500];
        char temp[500];
        
        int i = 1;
        while (true) 
        {
            if (i == num1) 
            {
                fgets(str1, 500, fin);
                cout << num1 << ") " << str1;
                ++i;
                break;
            }
            fgets(temp, 500, fin); 
            ++i;
        }
        while (true)
        {
            if (i == num2) 
            {
                fgets(str2, 500, fin);
                cout << num2 << ") " << str2;
                ++i;
                break;
            }
            fgets(temp, 500, fin); 
            ++i;
        }
    }
}
 
 
int main(int argc, char *argv[])
{
    //createfile();
    
    HANDLE file = CreateFile(filename, GENERIC_READ | GENERIC_WRITE, 0, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
    DWORD fileSize = GetFileSize(file, NULL);
    HANDLE fileMap =  CreateFileMapping(file, NULL, PAGE_READWRITE, 0, 0, NULL);
    CloseHandle(file);
    char *pvFileSrc = (char*)MapViewOfFile(fileMap, FILE_MAP_WRITE, 0, 0, fileSize);
    CloseHandle(fileMap);
    
    //cout << pvFileSrc; // вывод всех строк из проекции файла (демонстрация). В памяти, после последнего символа строки, стоят '\0',
                         // поэтому корректный вывод возможен
    
    int flag_n = 0;                                 // флаг наличия в конце последней строки '\n' (1, если есть)
    unsigned long c = 0;                            // счётчик байтов
    
    int n = numberStr(pvFileSrc, fileSize, flag_n); // количество строк
    cout << "n = " << n << endl << endl;
    
    int num1 = 4, num2 = 1;  // номера переставляемых строк
    
    if (num1 < 0 || num1 > n || num2 < 0 || num2 > n) // проверка корректности номеров строк
    {
        cout << "Incorrect line numbers: " << num1 << ", " << num2 << " !" << endl;
        return 0;
    }
    if (num1 > num2) // если номер первой строки больше второй, то перестановка значений
    {
        int temp = num1;
        num1 = num2;
        num2 = temp;
    }
    
    //showStr(num1, num2); // вывод двух строк из файла до перестановки (демонстрация)
    //cout << endl << endl;
    
    string str1, str2;
    
    char *num1_begin = findStr(pvFileSrc, num1, 1, c); // начало num1, первой из переставляемых строк (ищем от начала строк)
    char *num1_end   = findStr(num1_begin, 2,   1, c); // конец num1 (начало следующей)  
    
    str1.assign(num1_begin, num1_end); // сохраняем num1 (будем менять на num2)
    
    char *num2_begin = findStr(num1_end, num2, num1 + 1, c); // начало num2, второй из переставляемых строк (ищем от следующей за num1)
    char *num2_end;
    if (num2 == n && flag_n == 0) // если последняя строка не заканчивается символом новой строки
    {
        int size_end_str = fileSize - c;        // размер последней строки
        num2_end  = num2_begin + size_end_str;  // конец num2 (конец проекции файла)
    
    }
    else num2_end  = findStr(num2_begin, 2, 1, c);      // конец num2 (начало следующей) 
                                                   
    str2.assign(num2_begin, num2_end); // сохраняем num2 (будем менять на num1)
    
    int offset = str2.size() - str1.size(); // размер смещения данных между переставляемыми строками
 
    if ((offset == 0 && flag_n == 1) || (offset == 0 && num2 != n)) // если переставляемые строки равны 
                                                                    // и последняя строка заканчивается символом новой строки
                                                                    // или вторая строка не последняя
    {
        cout << num1 << ") " << str1;   // выводим переставляемые строки (демонстрация)     
        cout << num2 << ") " << str2;
        cout << endl;
        
        // переставляем строки
        strncpy(num1_begin, str2.c_str(), str2.size()); 
        strncpy(num2_begin, str1.c_str(), str1.size());
    }
    else if (num2 == n && flag_n == 0) // если последняя строка не заканчивается символом новой строки
         {
            str2.append("\r\n");
            if (str2.size() - str1.size() == 0)
            {
                str1.erase(str1.size() - 2, 2);
 
                // выводим переставляемые строки (демонстрация)
                cout << num1 << ") " << str1;        
                //if (num2 == n && flag_n == 0) // если последняя строка не заканчивается символом новой строки
                    cout << endl;
                cout << num2 << ") " << str2;
                cout << endl;
                
                // переставляем строки
                strncpy(num1_begin, str2.c_str(), str2.size()); 
                strncpy(num2_begin, str1.c_str(), str1.size());
            }
            else 
            {
                str1.erase(str1.size() - 2, 2);
                offset += 2; 
                
                // выводим переставляемые строки (демонстрация)
                cout << num1 << ") " << str1;     
                cout << endl;
                cout << num2 << ") " << str2;
                cout << endl;
 
                transfer(num1_end, num2_begin, offset);  // перемещаем данные между переставляемыми строками  
         
                // переставляем строки
                strncpy(num1_begin, str2.c_str(), str2.size());         
                strncpy(num2_begin + offset, str1.c_str(), str1.size());
            }
        }
        else
        {
            // выводим переставляемые строки (демонстрация)
            cout << num1 << ") " << str1;        
            if (num2 == n && flag_n == 0) // если последняя строка не заканчивается символом новой строки
                cout << endl;
            cout << num2 << ") " << str2;
            cout << endl;
 
            transfer(num1_end, num2_begin, offset);     // перемещаем данные между переставляемыми строками  
            
            // переставляем строки
            strncpy(num1_begin, str2.c_str(), str2.size()); 
            strncpy(num2_begin + offset, str1.c_str(), str1.size());
        }
            
    // демонстрация
    showStr(num1, num2);          // вывод двух строк из файла после перестановки
    if (num2 == n && flag_n == 0) // если последняя строка не заканчивается символом новой строки
        cout << endl;
    
    //cout << pvFileSrc;  // вывод всех строк после перестановки (демонстрация)
 
    UnmapViewOfFile( pvFileSrc );
    
    return 0;
}

// если одна из строк последняя, то проверить наличие '\n' в конце строки.
// если нет, то добавить, а у строки, которая будет последней - убрать '\n'
 
#include <iostream>
#include <string>
#include <ctime>
#include <fstream>
 
using namespace std;
 
const int N = 1000000;
 
void createfile() //создание файла для теста
{
    string str = "AAAAAAAAAA";
    ofstream fout("file1.txt");
    for (int i = 0; i < N; ++i)
    {
        fout << i + 1;
        fout << str << endl;
    }
    fout.close();
}
 
int main()
{
    //createfile(); //создание файла для теста
    
    char str1[255] = {'\0'};
    char str2[255] = {'\0'};
    char temp[255] = {'\0'};
    
    clock_t t1 = clock();
    FILE *fin = fopen("file1.txt", "rb");
    
    if (!fin) printf("Error!\n");
    else
    {
        int i = 1;
        int n = 1, m = 1000000; // номера заменяемых строк (не должны выходить за диапазон)
        while (true) 
        {
            if (i == n) 
            {
                fgets(str1, 255, fin);
                ++i;
            }
            if (i == m) 
            {
                fgets(str2, 255, fin);
                ++i;
            }
            fgets(temp, 255, fin); // предполагается, что длина строки не более 255 символов
            if (feof(fin)) break;
            ++i;
        }
        rewind(fin);
        
        printf("%d\n", i - 1);
        
        i = 0;
        char ch;
        FILE *fout = fopen("file2.txt", "wb");
        while (true)
        {
            fgets(temp, 255, fin);
            if (feof(fin)) break;
            ++i;
            if (i == n)  fputs(str2, fout);
            else if (i == m) fputs(str1, fout);
                 else  fputs(temp, fout);
        }
        
        fclose(fout);
        fclose(fin);
        
        clock_t t2 = clock();
        printf("%f\n", (t2 - t1 + .0) / CLOCKS_PER_SEC); // время отработки
    }
  
    system("pause");
    return 0;
}