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相关内容可查看本站专题：《Python字典操作技巧汇总》、《Python正则表达式用法总结》、《Python数据结构与算法教程》、《Python Socket编程技巧总结》、《Python函数使用技巧总结》、《Python字符串操作技巧汇总》、《Python入门与进阶经典教程》及《Python文件与目录操作技巧汇总》

希望本文所述对大家Python程序设计有所帮助。

本文标题: Python实现简单字典树的方法

本文地址: http://www.cppcns.com/jiaoben/python/148685.html