嗨,我正在编写我们自己的霍夫曼编码项目。我目前无法将二进制 1 和 0 写入输出文件。它适用于较小的输入文件,但对于非常大的文件,它不会将任何内容写入输出文件。负责写的方法就是compress
方法。任何帮助,将不胜感激。谢谢!
package proj3;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Collections;
import java.util.LinkedHashMap;
import java.util.Map;
import java.util.Scanner;
import java.util.PriorityQueue;
public class Project3 {
//variables for PriorityQueue and Huffman Tree
private static PriorityQueue<BinaryNode<Character>> queue;
private static BinaryNode<Character> huffTree;
private static Map<Character, String> table = new LinkedHashMap<Character, String>();
/**
* Method for creating Huffman Tree
* @param counts Map that contains all characters and their frequencies
* @return the Huffman Tree
*/
public static BinaryNode<Character> makeTree(Map<Character, Integer> counts)
{
queue = new PriorityQueue<BinaryNode<Character>>();
for(Character c : counts.keySet())
{
BinaryNode<Character> tree = new BinaryNode<Character>(c, counts.get(c), null, null);
queue.add(tree);
}
while(!queue.isEmpty())
{
if(queue.size() >= 2)
{
BinaryNode<Character> n1 = queue.remove();
BinaryNode<Character> n2 = queue.remove();
Integer weight = n1.getFreq() + n2.getFreq();
huffTree = new BinaryNode<Character>(null, weight, n1, n2);
queue.add(huffTree);
}
if(queue.size() == 1)
{
return queue.remove();
}
}
return huffTree;
}
public static void encode(BinaryNode<Character> node, String s)
{
if(!node.isLeaf())
{
encode(node.getLeft(), s + "0");
encode(node.getRight(), s + "1");
}
else
{
table.put(node.getElement(), s);
}
}
public static void compress(String in, String out) throws IOException
{
try
{
File outFile = new File(out);
FileOutputStream compressedFile = new FileOutputStream(outFile);
Scanner infile = new Scanner(new FileInputStream(in));
while(infile.hasNext())
{
infile.useDelimiter("");
String str = infile.next();
Character character = str.charAt(0);
for(Character c : table.keySet())
{
if(c == character){
compressedFile.write(table.get(c).getBytes());
compressedFile.flush();
}
}
}
for(Byte b : table.get('^').getBytes())
{
compressedFile.write(b);
}
infile.close();
compressedFile.close();
}
catch (FileNotFoundException e)
{
System.err.println("File not found.");
e.printStackTrace();
}
}
public static void decompress(String s)
{
}
public static void printEncodings(Map<Character, String> m)
{
ArrayList<Character> chars = new ArrayList<Character>();
System.out.println("Character Encodings");
System.out.println("---------------------");
for(Character c : m.keySet())
{
chars.add(c);
Collections.sort(chars);
}
for(Character c : chars)
{
System.out.print(c + "\t" + m.get(c) + "\n");
}
System.out.println();
System.out.println("Total Characters: " + chars.size());
}
/**
* Method for creating map with character and its frequencies
* @param s the file name to be opened
* @return the Map containing characters and frequencies
*/
public static Map<Character, Integer> charCount(String s){
Map<Character, Integer> counts = new LinkedHashMap<Character, Integer>();
ArrayList<Character> chars = new ArrayList<Character>();
try {
Scanner file = new Scanner(new FileInputStream(s));
while(file.hasNext()){
file.useDelimiter("");
String str = file.next();
Character c = str.charAt(0);
if(counts.containsKey(c)){
counts.put(c, counts.get(c) + 1);
}
else{
counts.put(c, 1);
}
}
counts.put('^', 1);
System.out.println("Character Frequencies");
System.out.println("---------------------");
for(Character c : counts.keySet())
{
chars.add(c);
Collections.sort(chars);
}
for(Character c : chars){
System.out.println(c + "\t" + counts.get(c));
}
System.out.println();
System.out.println("Total characters: " + chars.size() + "\n");
file.close();
}
catch (FileNotFoundException e) {
System.err.println("File not found.");
System.exit(0);
}
return counts;
}
public static void main(String[] args){
if(args.length != 3)
{
throw new IllegalArgumentException("Invalid number of arguments.");
}
encode(makeTree(charCount(args[0])), "");
printEncodings(table);
try {
compress(args[0], args[1]);
} catch (IOException e) {
e.printStackTrace();
}
}
}