数据结构（6）-JAVA-哈夫曼树的构造及显示

掌握如何构造哈夫曼树并显示哈夫曼树

实验任务：

已知7个叶子的权值为{7,5,2,3,8,10,20}，设计算法构造哈夫曼树并显示在控制台。

项目结构：

程序设计思路：

哈夫曼结点类型定义HuffmanNode(权值weight,左子树lchild，右子树rchild，双亲parent
定义哈夫曼树类Huffman：
初始化，写入权值init()
构造哈夫曼树huffman()
输出哈夫曼结点的值display()
用主方法main()调用功能方法
调试源代码。

哈夫曼树结点类型HuffmanNode:

package ds.huffman;
/**
 * 哈夫曼树结点类型
 * @author 1
 *
 */
public class HuffmanNode {
//权值
 private int weight;
 //双亲左孩子右孩子
 private int parent,lchild,rchild;

//构造方法
 public HuffmanNode() {
  super();
  this.weight = 0;
  this.parent = 0;
  this.lchild = 0;
  this.rchild = 0;
 }
 public HuffmanNode(int weight, int parent, int lchild, int rchild) {
  super();
  this.weight = weight;
  this.parent = parent;
  this.lchild = lchild;
  this.rchild = rchild;
 }

 //set和get方法
 public int getWeight() {
  return weight;
 }
 public void setWeight(int weight) {
  this.weight = weight;
 }
 public int getParent() {
  return parent;
 }
 public void setParent(int parent) {
  this.parent = parent;
 }
 public int getLchild() {
  return lchild;
 }
 public void setLchild(int lchild) {
  this.lchild = lchild;
 }
 public int getRchild() {
  return rchild;
 }
 public void setRchild(int rchild) {
  this.rchild = rchild;
 }
}

哈夫曼类Huffman:

package ds.huffman;

import java.util.Scanner;

/**
 * 哈夫曼树
 * 
 * @author 1
 * 
 */
public class Huffman {
 // 数组存放哈夫曼树中的每个结点信息
 private HuffmanNode[] huf;

// 构造方法：数组的初始化操作
 public Huffman(int n) {// 已知有n个叶子结点
  huf = new HuffmanNode[2 * n - 1];
  // 初始化数组 0~length 设置为0
  Scanner in = new Scanner(System.in);
  int i;
  for (i = 0; i < huf.length; i++) {
   huf[i] = new HuffmanNode();
   // 0~n 结点的权值写入
   if (i < n) {
    System.out.println("请输入第" + (i + 1) + "个叶子的权值：");
    huf[i].setWeight(in.nextInt());
   }
  }
 }

// 构造哈夫曼树 2n-1=length
 public void huffman() {
  for (int i = (huf.length + 1) / 2; i < huf.length; i++) {// 新结点的存放位置
   int min1 = 999, min2 = 999;
   int p1 = -1, p2 = -1;
   for (int j = 0; j < i; j++) {// 查找最小权值和次小权值
    if (huf[j].getParent() == 0) {
     if (huf[j].getWeight() < min1) {// 是否小于min1，最小值
      min2 = min1;
      p2 = p1;
      min1 = huf[j].getWeight();
      p1 = j;
     } else if (huf[j].getWeight() < min2) {// 是否小于min2，次小值
      min2 = huf[j].getWeight();
      p2 = j;
     }
    }
   }
   // 存入新结点的数据
   huf[i].setWeight(min1 + min2);
   huf[i].setLchild(p1 + 1);
   huf[i].setRchild(p2 + 1);
   huf[p1].setParent(i + 1);// 叶子结点
   huf[p2].setParent(i + 1);
  }
 }

// 显示哈夫曼树
 public void show() {
  System.out.println("序号\t\t左孩子序号\t\t右孩子序号\t\t双亲序号\t\t结点权值");
  for (int i = 0; i < huf.length; i++) {
   System.out.println((i + 1) + "\t\t" + huf[i].getLchild() + "\t\t"
     + huf[i].getRchild() + "\t\t" + huf[i].getParent() + "\t\t"
     + huf[i].getWeight());
  }
 }
}

test:

public class Test {
 public static void main(String[] args) {
  Huffman hf = new Huffman(7);
  hf.huffman();
  hf.show();
 }
}

运行截图：

该文章仅供自己学习和参考，如果有错误的地方希望大家能够指正！