Trie后缀树Python简单实现

最新推荐文章于 2023-01-12 15:22:27 发布

weixin_34253539

最新推荐文章于 2023-01-12 15:22:27 发布

阅读量1.1k

点赞数

CC 4.0 BY-SA版权

文章标签： python

原文链接：https://my.oschina.net/leopardsaga/blog/96727

本文介绍了一种使用Python实现Trie树的方法，并提供了一个具体的示例。通过该示例，读者可以了解到如何创建Trie树并对其进行序列计数。此外，还介绍了如何将构建好的Trie树序列化为文件。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

2019独角兽企业重金招聘Python工程师标准>>>

#!/usr/bin/env python
# -*- coding: utf-8 -*-
'''
Created on Dec 17, 2012


@author: honghe
'''


import pickle


class TrieNode(object):
    def __init__(self):
        self.count = 1  # 统计此结点代表的字符串出现的次数
        self.children = {}  
        
class Trie(object):
    def __init__(self):
        self.root = TrieNode()
        
    def add(self, sequence):
        node = self.root
        for c in sequence:
            if c not in node.children:
                child = TrieNode()
                node.children[c] = child
                node = child
            else:
                node = node.children[c]
                node.count = node.count + 1
    
    def countSeq(self, sequence):
        '''计算序列出现的次数
        '''
        node = self.root
        for c in sequence:
            if c not in node.children:
                return 0
            else:
                node = node.children[c]
        return node.count
    
def gen_trie(input_file, output_file):
    '''生成trie树
    '''
    trie = Trie()
    
    with open(input_file) as f:
        for line in f:
            # 增加'$'用来区别是否是完整后缀
            line = line.strip() + '$'
            for i in range(len(line)):
                l = line[i:]
                trie.add(l)
            
    with open(output_file, 'wb') as f:
        pickle.dump(trie, f)
        
    return trie
  
if __name__ == '__main__':
    txt = 'data.txt'
    pkl = 'data.pkl'
    t = gen_trie(txt, pkl)

转载于:https://my.oschina.net/leopardsaga/blog/96727