霍夫曼代码将位写入文件以进行压缩

时间:2017-04-15 01:36:48

标签: java algorithm compression huffman-code greedy

我被要求使用huffman代码压缩输入文件并将其写入输出文件。我已经完成了霍夫曼树结构的实现并生成了霍夫曼代码。但我不知道如何将这些代码写入文件,以使文件的大小小于原始文件。

现在我的代码是字符串表示(例如' c'的霍夫曼代码是" 0100")。有人请帮我写那些比特  文件。

1 个答案:

答案 0 :(得分:3)

这里有一种将比特流(霍夫曼编码的输出)写入文件的可能实现。

class BitOutputStream {

    private OutputStream out;
    private boolean[] buffer = new boolean[8];
    private int count = 0;

    public BitOutputStream(OutputStream out) {
        this.out = out;
    }

    public void write(boolean x) throws IOException {
        this.count++;
        this.buffer[8-this.count] = x;
        if (this.count == 8){
            int num = 0;
            for (int index = 0; index < 8; index++){
                num = 2*num + (this.buffer[index] ? 1 : 0);
            }

            this.out.write(num - 128);

            this.count = 0;
        }
    }

    public void close() throws IOException {
        int num = 0;
        for (int index = 0; index < 8; index++){
            num = 2*num + (this.buffer[index] ? 1 : 0);
        }

        this.out.write(num - 128);

        this.out.close();
    }

}

通过调用write方法,您可以在文件(OutputStream)中逐位写入。

修改

针对您的具体问题,要保存每个角色的霍夫曼代码,如果您不想使用其他一些花哨的类,您可以使用此代码 -

String huffmanCode = "0100"; // lets say its huffman coding output for c

BitSet huffmanCodeBit = new BitSet(huffmanCode.length());

for (int i = 0; i < huffmanCode.length(); i++) {
    if(huffmanCode.charAt(i) == '1')
        huffmanCodeBit.set(i);
}
String path = Resources.getResource("myfile.out").getPath();
ObjectOutputStream outputStream = null;
try {
    outputStream = new ObjectOutputStream(new FileOutputStream(path));
    outputStream.writeObject(huffmanCodeBit);
} catch (IOException e) {
    e.printStackTrace();
}