Python完成哈夫曼樹編碼過程及原理詳解

更新時間：2019年07月29日 08:52:53 作者：TomHawk

這篇文章主要介紹了Python完成哈夫曼樹編碼過程及原理詳解,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下

哈夫曼樹原理

秉著能不寫就不寫的理念，關于哈夫曼樹的原理及其構建，還是貼一篇博客吧。

http://www.dbjr.com.cn/article/97396.htm

其大概流程

哈夫曼編碼代碼

# 樹節(jié)點類構建
class TreeNode(object):
  def __init__(self, data):
    self.val = data[0]
    self.priority = data[1]
    self.leftChild = None
    self.rightChild = None
    self.code = ""
# 創(chuàng)建樹節(jié)點隊列函數(shù)
def creatnodeQ(codes):
  q = []
  for code in codes:
    q.append(TreeNode(code))
  return q
# 為隊列添加節(jié)點元素，并保證優(yōu)先度從大到小排列
def addQ(queue, nodeNew):
  if len(queue) == 0:
    return [nodeNew]
  for i in range(len(queue)):
    if queue[i].priority >= nodeNew.priority:
      return queue[:i] + [nodeNew] + queue[i:]
  return queue + [nodeNew]
# 節(jié)點隊列類定義
class nodeQeuen(object):

  def __init__(self, code):
    self.que = creatnodeQ(code)
    self.size = len(self.que)

  def addNode(self,node):
    self.que = addQ(self.que, node)
    self.size += 1

  def popNode(self):
    self.size -= 1
    return self.que.pop(0)
# 各個字符在字符串中出現(xiàn)的次數(shù)，即計算優(yōu)先度
def freChar(string):
  d ={}
  for c in string:
    if not c in d:
      d[c] = 1
    else:
      d[c] += 1
  return sorted(d.items(),key=lambda x:x[1])
# 創(chuàng)建哈夫曼樹
def creatHuffmanTree(nodeQ):
  while nodeQ.size != 1:
    node1 = nodeQ.popNode()
    node2 = nodeQ.popNode()
    r = TreeNode([None, node1.priority+node2.priority])
    r.leftChild = node1
    r.rightChild = node2
    nodeQ.addNode(r)
  return nodeQ.popNode()

codeDic1 = {}
codeDic2 = {}
# 由哈夫曼樹得到哈夫曼編碼表
def HuffmanCodeDic(head, x):
  global codeDic, codeList
  if head:
    HuffmanCodeDic(head.leftChild, x+'0')
    head.code += x
    if head.val:
      codeDic2[head.code] = head.val
      codeDic1[head.val] = head.code
    HuffmanCodeDic(head.rightChild, x+'1')
# 字符串編碼
def TransEncode(string):
  global codeDic1
  transcode = ""
  for c in string:
    transcode += codeDic1[c]
  return transcode
# 字符串解碼
def TransDecode(StringCode):
  global codeDic2
  code = ""
  ans = ""
  for ch in StringCode:
    code += ch
    if code in codeDic2:
      ans += codeDic2[code]
      code = ""
  return ans
# 舉例
string = "AAGGDCCCDDDGFBBBFFGGDDDDGGGEFFDDCCCCDDFGAAA"
t = nodeQeuen(freChar(string))
tree = creatHuffmanTree(t)
HuffmanCodeDic(tree, '')
print(codeDic1,codeDic2)
a = TransEncode(string)
print(a)
aa = TransDecode(a)
print(aa)
print(string == aa)

接下來就是一段一段分析代碼

1.樹結點類的構建：

共有5個屬性：結點的值，結點的優(yōu)先度，結點的左子結點，結點的右子結點，結點值的編碼（這個沒有什么好說的，這些屬性都是被需要的）

2.創(chuàng)建樹結點隊列函數(shù)：

對于所有的字母結點，我們將其組成一個隊列，這里使用list列表來完成隊列的功能。將所有樹節(jié)點夠放進列表中，當然傳進來的是按優(yōu)先度從小到大已排序的元素列表

3.為隊列添加節(jié)點元素，并保證優(yōu)先度從大到小排列：

當有新生成的結點時，需將其插入列表，并放在合適位置，使隊列依然時按優(yōu)先度從小打到排列的。

4.結點隊列類定義：

創(chuàng)建類初始化時需要傳進去的是一個列表，列表中的每個元素是由字母與優(yōu)先度組成的元組。元組第一個元素是字母，第二個元素是優(yōu)先度（即在文本中出現(xiàn)的次數(shù)）

類初始化化時，調用“創(chuàng)建樹結點隊列函數(shù)”，隊列中的每個元素都是一個樹結點。

類中還包含一個隊列規(guī)模屬性以及另外兩個操作函數(shù)：添加結點函數(shù)和彈出結點函數(shù)。

添加結點函數(shù)直接調用之前定義的函數(shù)即可，輸入的參數(shù)為隊列和新結點，并且隊列規(guī)模加一