在数据结构书中,讲解树内容的时候,都会介绍哈夫曼树(Huffman)和哈夫曼编码(哈夫曼树的一种应用)。关于哈夫曼树的定义,在这里就不讨论了,接下来贴出LZ实现哈夫曼树的一种方案。
构建哈夫曼树:
- 1.首先将所有的节点构成独立的二叉树,这些二叉树构成的一个森林,将这些二叉树构建成一个最小堆;
- 2.选择最小堆中两个最小树,构建成一颗新的树,将新树插入到最小堆中,删除被选出的两颗最小树;
- 3.如果最小堆节点数大于1,重复2。
- 4.最小堆中最后一个节点,这棵树就是哈夫曼树。
本文中二叉树的实现类似二叉排序树,最小堆的实现如数据结构—堆。构建哈夫曼树的实现如下:
1 #define MaxSize 100 2 template<typename Type> 3 void Huffman(Type *elements, int n, BinaryTree<Type> &tree){ 4 BinaryTree<Type> first, second; 5 BinaryTree<Type> node[MaxSize]; 6 for (int i=0; i<n; i++){ 7 node[i].m_proot = new BinTreeNode<Type>(elements[i]); 8 } 9 MinHeap<BinaryTree<Type> > heap(node, n); 10 11 for (int i=0; i<n-1; i++){ 12 frist=heap.DeleteMin(); 13 second=heap.DeleteMin(); 16 if (first.m_proot->GetData() == second.m_proot->GetData()){ 17 tree = *(new BinaryTree<Type>(second, first)); 18 } 19 else { 20 tree = *(new BinaryTree<Type>(first, second)); 21 } 22 23 heap.Insert(tree); 24 } 25 }