[英]Storing word count in the python trie
I took a list of words and put it into a trie.我拿了一个单词列表,把它放在一个特里。 I would also like to store word count inside for further analysis.我还想在里面存储字数以供进一步分析。 What would be the best way to do it?最好的方法是什么? This is the class where I think the frequency would be collected and stored, but I am not sure how to go about it.这是我认为频率将被收集和存储的类,但我不知道如何去做。 You can see my attempt, last line in insert is where I try to store the count.你可以看到我的尝试,插入中的最后一行是我尝试存储计数的地方。
class TrieNode:
def __init__(self,k):
self.v = 0
self.k = k
self.children = {}
def all_words(self, prefix):
if self.end:
yield prefix
for letter, child in self.children.items():
yield from child.all_words(prefix + letter)
class Trie:
def __init__(self):
self.root = TrieNode()
def __init__(self):
self.root = TrieNode()
def insert(self, word):
curr = self.root
for letter in word:
node = curr.children.get(letter)
if not node:
node = TrieNode()
curr.children[letter] = node
curr.v += 1
def insert_many(self, words):
for word in words:
self.insert(word)
def all_words_beginning_with_prefix(self, prefix):
cur = self.root
for c in prefix:
cur = cur.children.get(c)
if cur is None:
return # No words with given prefix
yield from cur.all_words(prefix)
I want to store the count so that when I use我想存储计数,以便在使用时
print(list(trie.all_words_beginning_with_prefix('prefix')))
I would get a result like so:我会得到这样的结果:
[(word, count), (word, count)]
While inserting , on seeing any node, it means there's a new word going to be added in that path.插入时,在看到任何节点时,这意味着将在该路径中添加一个新词。 Therefore increment your word_count of that node.因此,增加该节点的 word_count。
class TrieNode:
def __init__(self, char):
self.char = char
self.word_count = 0
self.children = {}
def all_words(self, prefix, path):
if len(self.children) == 0:
yield prefix + path
for letter, child in self.children.items():
yield from child.all_words(prefix, path + letter)
class Trie:
def __init__(self):
self.root = TrieNode('')
def insert(self, word):
curr = self.root
for letter in word:
node = curr.children.get(letter)
if node is None:
node = TrieNode(letter)
curr.children[letter] = node
curr.word_count += 1 # increment it everytime the node is seen at particular level.
curr = node
def insert_many(self, words):
for word in words:
self.insert(word)
def all_words_beginning_with_prefix(self, prefix):
cur = self.root
for c in prefix:
cur = cur.children.get(c)
if cur is None:
return # No words with given prefix
yield from cur.all_words(prefix, path="")
def word_count(self, prefix):
cur = self.root
for c in prefix:
cur = cur.children.get(c)
if cur is None:
return 0
return cur.word_count
trie = Trie()
trie.insert_many(["hello", "hi", "random", "heap"])
prefix = "he"
words = [w for w in trie.all_words_beginning_with_prefix(prefix)]
print("Lazy method:\n Prefix: %s, Words: %s, Count: %d" % (prefix, words, len(words)))
print("Proactive method:\n Word count for '%s': %d" % (prefix, trie.word_count(prefix)))
Output:输出:
Lazy method:
Prefix: he, Words: ['hello', 'heap'], Count: 2
Proactive method:
Word count for 'he': 2
I would add a field called is_word to the trie node, where is_word would be true only for the last letter in the word.我会向 trie 节点添加一个名为 is_word 的字段,其中 is_word 仅对单词中的最后一个字母为真。 Like you have word AND, is_word would be true for the trie node holding the letter D. And I would update frequency for only nodes that have is_word to be true, not for every letter in the word.就像你有单词 AND 一样,对于包含字母 D 的 trie 节点,is_word 将是真的。我只会更新具有 is_word 的节点的频率,而不是单词中的每个字母。
So when you iterate from a letter, check if it is a word, if it is, stop the iteration, return the count and the word.所以当你从一个字母迭代时,检查它是否是一个单词,如果是,停止迭代,返回计数和单词。 I'm assuming in your iteration you keep track of the letters, and keep adding them to the prefix.我假设在您的迭代中您会跟踪这些字母,并不断将它们添加到前缀中。
Your trie is a multi-way trie.你的特里是一个多路特里。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.