C++中霍夫曼编码后变长比特串写入二进制文件的实现方法

首先直接回答你的核心疑问：std::bitset并不适合你的场景。因为bitset是编译期固定大小的模板类，必须在代码里指定具体长度（比如bitset<2611>），但你的比特串长度是动态变化的，甚至可能非常长，这显然无法满足需求。我们需要用动态的方式来处理变长比特串。

下面是具体的实现步骤和代码示例：

核心思路 #

1. 确保比特串长度为8的倍数：你已经提到会做填充，这里要注意：如果后续需要解压，一定要把填充的位数记录到文件中（比如写在文件开头的1个字节里），否则解压时无法区分原始数据和填充位。
2. 逐组转换比特串为字节：把每8位比特串转换为一个uint8_t（无符号字节），然后以二进制模式写入文件。
3. 二进制模式写入：必须用std::ios::binary打开文件，避免操作系统对换行符等字符的自动转换，破坏二进制数据。

代码实现（针对你的示例） #

假设你的比特串存储在std::string中（用'0'和'1'表示每一位），下面的代码可以直接将其转为二进制文件：

#include <fstream>
#include <string>
#include <cstdint>
#include <stdexcept>
#include <iostream>

void write_huffman_bits_to_binary(const std::string& bitstring, const std::string& output_path) {
    // 先校验比特串长度是否为8的倍数
    if (bitstring.size() % 8 != 0) {
        throw std::invalid_argument("Bitstring length must be a multiple of 8 after padding.");
    }

    // 以二进制模式打开输出文件
    std::ofstream out_file(output_path, std::ios::binary);
    if (!out_file.is_open()) {
        throw std::runtime_error("Failed to open output file.");
    }

    // 每8位转换为一个字节
    for (size_t i = 0; i < bitstring.size(); i += 8) {
        uint8_t current_byte = 0;
        // 从左到右处理每一位，对应字节的高位到低位
        for (size_t j = 0; j < 8; ++j) {
            current_byte <<= 1; // 左移一位，腾出位置给下一位
            if (bitstring[i + j] == '1') {
                current_byte |= 1; // 如果是1，设置当前最低位为1
            }
            // 如果是0，保持当前位为0即可
        }
        // 将字节写入文件
        out_file.write(reinterpret_cast<const char*>(&current_byte), sizeof(current_byte));
    }

    out_file.close();
}

// 测试你的示例比特串
int main() {
    std::string sample_bits = "10010101010010101010101011101100001001000101011110110101001010101001001010110101";
    // 你的示例长度是80，刚好是8的倍数，无需额外填充
    try {
        write_huffman_bits_to_binary(sample_bits, "compressed.bin");
        std::cout << "Binary file written successfully!" << std::endl;
    } catch (const std::exception& e) {
        std::cerr << "Error: " << e.what() << std::endl;
        return 1;
    }
    return 0;
}

关键细节说明 #

• 位顺序处理：代码中默认比特串的第一个字符是字节的最高位（比如示例中的前8位10010101会被转为0x95，即十进制149）。如果你的霍夫曼编码生成的比特串是低位在前，只需要调整内层循环的顺序（从j=7到j=0，或者改为右移操作）。
• 动态长度支持：用std::string存储比特串可以轻松支持任意长度（只要内存足够），完全适配你的变长需求。
• 填充注意事项：比如原始比特串长度是2611，你需要填充5位（2611+5=2616，8的倍数）。建议填充的是0，并把填充的位数（5）写入文件开头，这样解压时可以准确截断多余的位。

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