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

#coding=utf8

"""代碼實現了最簡單的字典樹，只支持由小寫字母組成的字符串。

在此代碼基礎上擴展一下，就可以實現比較復雜的字典樹，比如帶統計數的，或支持更多字符的字典樹，

或者是支持刪除等操作。

"""

class TrieNode(object):

def __init__(self):

# 是否構成一個完成的單詞

self.is_word = False

self.children = [None] * 26

class 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 False

if __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树的实现_Python实现简单字典树的方法的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。