java哈夫曼编码译码_哈夫曼编码与译码 java

展开全部

package qwp;

import java.util.ArrayDeque;

import java.util.ArrayList;

import java.util.Collections;

import java.util.HashSet;

import java.util.List;

import java.util.Queue;

import java.util.Scanner;

public class HuffmanCoding {

public static String writeString;

public static class HNode {

String data = "";

String coding = "";

@Override

public String toString() {

return "HNode [coding=" + coding + ", data=" + data + "]";

}

public HNode(String data) {

super();

this.data = data;

}

@Override

public int hashCode() {

final int prime = 31;

int result = 1;

result = prime * result + ((data == null) ? 0 : data.hashCode());

return result;

}

@Override

public boolean equals(Object obj) {

if (this == obj)

return true;

if (obj == null)

return false;

if (getClass() != obj.getClass())

return false;

HNode other = (HNode) obj;

if (data == null) {

if (other.data != null)

return false;

} else if (!data.equals(other.data))

return false;

return true;

}

}

public static class Node {

HNode data;

int weight;

Node leftChild;

Node rightChild;

public Node(HNode data, int weight) {

this.data = data;

this.weight = weight;

}

public String toString() {

return "Node[data=" + data + ", weight=" + weight + "]";

}

@Override

public int hashCode() {

final int prime = 31;

int result = 1;

result = prime * result + ((data == null) ? 0 : data.hashCode());

return result;

}

@Override

public boolean equals(Object obj) {

if (this == obj)

return true;

if (obj == null)

return false;

if (getClass() != obj.getClass())

return false;

Node other = (Node) obj;

if (data == null) {

if (other.data != null)

return false;

} else if (!data.equals(other.data))

return false;

return true;

}

}

public static void main(String[] args) {

System.out.println("请输入字符串：");

Scanner scanner = new Scanner(System.in);

HuffmanCoding.writeString = scanner.nextLine();

char[] chars = writeString.toCharArray();

List nodes = new ArrayList();

for (int i = 0; i 

Node t = new Node(new HNode(String.valueOf(chars[i])), 1);

if (nodes.contains(t)) {

nodes.get(nodes.indexOf(t)).weight++;

} else {

nodes.add(t);

}

}

// System.out.println(nodes);

Node root = HuffmanCoding.createTree(nodes);

breadthFirst(root, nodes);

for (int i = 0; i 

Node t = new Node(new HNode(String.valueOf(chars[i])), 1);

System.out.print(nodes.get(nodes.indexOf(t)).data.coding);

}

}

private static Node createTree(List nodess) {

List nodes = new ArrayList(nodess);

// 只要nodes数组中还有2个以上的节点

while (nodes.size() > 1) {

quickSort(nodes);

// 获取权值最小的e5a48de588b662616964757a686964616f31333361326264两个节点

Node left = nodes.get(nodes.size() - 1);

Node right = nodes.get(nodes.size() - 2);

// 生成新节点，新节点的权值为两个子节点的权值之和

Node parent = new Node(new HNode(null), left.weight + right.weight);

// 让新节点作为权值最小的两个节点的父节点

parent.leftChild = left;

parent.rightChild = right;

// 删除权值最小的两个节点

nodes.remove(nodes.size() - 1);

nodes.remove(nodes.size() - 1);

// 将新生成的父节点添加到集合中

nodes.add(parent);

}

// 返回nodes集合中唯一的节点，也就是根节点

return nodes.get(0);

}

public static void quickSort(List nodes) {

subSort(nodes, 0, nodes.size() - 1);

}

private static void subSort(List nodes, int start, int end) {

if (start 

Node base = nodes.get(start);

int i = start;

int j = end + 1;

while (true) {

while (i = base.weight)

;

while (j > start && nodes.get(--j).weight <= base.weight)

;

if (i 

swap(nodes, i, j);

} else {

break;

}

}

swap(nodes, start, j);

// 递归左子序列

subSort(nodes, start, j - 1);

// 递归右边子序列

subSort(nodes, j + 1, end);

}

}

private static void swap(List nodes, int i, int j) {

Node tmp;

tmp = nodes.get(i);

nodes.set(i, nodes.get(j));

nodes.set(j, tmp);

}

// 广度优先遍历

public static void breadthFirst(Node root, List nodes) {

Queue queue = new ArrayDeque();

List list = new ArrayList();

if (root != null) {

// 将根元素入“队列”

queue.offer(root);

}

while (!queue.isEmpty()) {

// 将该队列的“队头”的元素添加到List中

list.add(queue.peek());//返回队列头部的元素

Node p = queue.poll();//

// 如果左子节点不为null，将它加入“队列”

if (p.leftChild != null) {

queue.offer(p.leftChild);

p.leftChild.data.coding = p.data.coding + "0";

} else {

((Node) nodes.get(nodes.indexOf(p))).data.coding = p.data.coding;

}

// 如果右子节点不为null，将它加入“队列”

if (p.rightChild != null) {

queue.offer(p.rightChild);

p.rightChild.data.coding = p.data.coding + "1";

}

}

}

}