霍夫曼编码文件解压剩余比特位处理问题求助

你遇到的问题核心是压缩阶段未记录有效比特总数，导致解压时误把最后一个字节的补位0当成有效比特处理，干扰Huffman树遍历逻辑，最终引发字符丢失。

解决方案：在压缩文件中写入原始编码的总比特数，解压时先读取该数值，仅处理对应数量的有效比特，忽略最后一个字节的补位无效比特。

1. 修改压缩代码（compressFile函数） #

新增总比特数统计并写入文件头部，同时修正文件打开模式为二进制（避免文本模式的换行转换问题）

void compressFile(string inputFile) {
    huffmanTree();
    system("cls");
    cout << "\n\n\t\t\t\tProcessing...";
    Sleep(5000);

    // 修正：以二进制模式打开输入文件
    ifstream inputedFile(inputFile, ios::binary);
    ofstream compressedFile("compressed.huff", ios::binary);

    if (!inputedFile.is_open() || !compressedFile.is_open()) {
        cout << "\t\t\t\tError: Unable to open file for compression." << endl;
        return;
    }

    string bits;
    char ch;
    int totalBits = 0; // 新增：统计所有有效编码的总比特数

    while (inputedFile.get(ch)) {
        string code = treeCode[(int)ch];
        bits += code;
        totalBits += code.length(); // 累计总比特数

        if(bits.length() >= 8){
            // 处理完整的8位组
            for (int i = 0; i + 8 <= bits.length(); i += 8) {
                compressedFile.put((char)stoi(bits.substr(i, 8), NULL, 2));
            }
            bits = bits.substr(bits.length() - bits.length() % 8);
        }
    }

    // 先写入总比特数到文件头部（4字节整数，支持最大约2GB原始数据）
    compressedFile.write(reinterpret_cast<const char*>(&totalBits), sizeof(totalBits));

    // 写入剩余不足8位的比特
    if (!bits.empty()) {
        compressedFile.put((char)stoi(bits, NULL, 2));
    }

    system("cls");
    cout << "\t\t\t\t---------------------------------------------" << endl;
    cout << "\n\n\t\t\t\tSuccessful: File has been compressed." << endl;
    cout << "\n\n\t\t\t\tThe file name is compressed.huff." << endl;
    cout << "\t\t\t\t---------------------------------------------" << endl;
    cout << "\t\t\t\t";
    system("pause");
    inputedFile.close();
    compressedFile.close();
}

2. 修改解压代码（decompressFile函数） #

先读取总比特数，严格控制有效比特的处理数量，避免处理补位无效比特

void decompressFile(string compressedFile) {
    system("cls");
    cout << "\n\n\t\t\t\tProcessing...";
    Sleep(5000);

    ifstream compressedFileStream(compressedFile, ios::binary);
    ofstream decompressedFile("decompressed.txt", ios::binary);

    if (!compressedFileStream.is_open() || !decompressedFile.is_open()) {
        cout << "\n\t\t\t\tError: Unable to open file for decompression." << endl;
        return;
    }

    huffmanTree();

    Node* root = head->node;
    Node* current = root;

    int totalBits;
    // 先读取文件头部的总比特数
    compressedFileStream.read(reinterpret_cast<char*>(&totalBits), sizeof(totalBits));

    char byte;
    int processedBits = 0; // 记录已处理的有效比特数

    while (compressedFileStream.get(byte) && processedBits < totalBits) {
        for (int i = 7; i >= 0 && processedBits < totalBits; i--) {
            processedBits++;
            char bit = (byte & (1 << i)) ? '1' : '0';

            if (bit == '0') {
                current = current->left;
            } else if (bit == '1') {
                current = current->right;
            }

            // 到达叶子节点，输出字符并重置遍历指针
            if (current->left == NULL && current->right == NULL) {
                decompressedFile << current->character;
                cout << "decompressed" << current->character;
                current = root;
            }
        }
    }

    system("pause");
    system("cls");
    cout << "\t\t\t\t---------------------------------------------" << endl;
    cout << "\n\n\t\t\t\tSuccessful: File has been decompressed." << endl;
    cout << "\n\n\t\t\t\tThe file name is decompressed.txt." << endl;
    cout << "\t\t\t\t---------------------------------------------" << endl;
    cout << "\t\t\t\t";
    system("pause");
    compressedFileStream.close();
    decompressedFile.close();
}

关键修改说明 #

1. 压缩阶段：

   • 新增totalBits统计所有Huffman编码的总长度，确保解压时明确知道需要处理的比特数量
   • 将totalBits写入压缩文件头部，用4字节整数存储，覆盖绝大多数文件场景
   • 修正输入输出文件为二进制打开模式，避免文本模式下的换行符转换破坏编码数据

2. 解压阶段：

   • 先读取头部的totalBits，用processedBits计数已处理的有效比特
   • 每次处理比特前判断processedBits < totalBits，仅处理有效比特，自动忽略最后一个字节的补位0
   • 输出文件改为二进制模式，避免文本模式的换行转换导致解压内容异常

内容的提问来源于stack exchange，提问作者Pewpew