Python實現簡單字典樹的方法

本文執行個體講述了Python實現簡單字典樹的方法。分享給大家供大家參考，具體如下：

#coding=utf8"""代碼實現了最簡單的字典樹，只支援由小寫字母組成的字串。在此代碼基礎上擴充一下，就可以實現比較複雜的字典樹，比如帶統計數的，或支援更多字元的字典樹，或者是支援刪除等操作。"""class TrieNode(object):  def __init__(self):    # 是否構成一個完成的單詞    self.is_word = False    self.children = [None] * 26class Trie(object):  def __init__(self):    self.root = TrieNode()  def add(self, s):    """Add a string to this trie."""    p = self.root    n = len(s)    for i in range(n):      if p.children[ord(s[i]) - ord('a')] is None:        new_node = TrieNode()        if i == n - 1:          new_node.is_word = True        p.children[ord(s[i]) - ord('a')] = new_node        p = new_node      else:        p = p.children[ord(s[i]) - ord('a')]        if i == n - 1:          p.is_word = True          return  def search(self, s):    """Judge whether s is in this trie."""    p = self.root    for c in s:      p = p.children[ord(c) - ord('a')]      if p is None:        return False    if p.is_word:      return True    else:      return Falseif __name__ == '__main__':  trie = Trie()  trie.add('str')  trie.add('acb')  trie.add('acblde')  print trie.search('acb')  print trie.search('ac')  trie.add('ac')  print trie.search('ac')

更多關於Python相關內容可查看本站專題：《Python字典操作技巧匯總》、《PythonRegex用法總結》、《Python資料結構與演算法教程》、《Python Socket編程技巧總結》、《Python函數提示總結》、《Python字串操作技巧匯總》、《Python入門與進階經典教程》及《Python檔案與目錄操作技巧匯總》

希望本文所述對大家Python程式設計有所協助。

