天天看點

python樹的實作_Python實作簡單字典樹的方法

本文執行個體講述了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程式設計有所幫助。