将Trie树节点作为参数传递-不起作用

Question

我正在尝试实现一个功能，该功能读取Trie树中包含的所有单词，将它们与键一起存储在列表中，并将它们写入.csv文件中。 函数“ insertTrie”工作正常； 但是，当我将“ root”作为参数传递给函数“ getAllTrie”时，由于某种原因，当我在函数中打印它（作为测试）时，它会向节点添加一个字符串（“ q”），然后显示“ AttributeError” ：'str'对象没有属性'char'”。 当我在函数外部打印“ root”时，字符串不存在。 是什么原因造成的？ 我花了很长时间试图找到答案。

import csv

class Node():
   def __init__(self):
       self.sons = {}
       self.char = None
       self.value = None

def insertTrie(currentNode, word, size):
    if(size == len(word)):
        if(currentNode.value is None):
        currentNode.value = 1
        else:
            currentNode.value += 1
        return currentNode
    currentChar = word[size]
    if(currentChar not in currentNode.sons):
        currentNode.sons[currentChar] = Node()
        currentNode.sons[currentChar].char = currentChar
    currentNode.sons[currentChar] = insertTrie(currentNode.sons[ccurrentChar], word, size+1)
    return currentNode

def getAllTrie(currentNode, findWord):
    if(currentNode is not None):
        #print(currentNode) -> print root inside function, 'q' appears
        if(currentNode.char is not None):
            if(findWord is None):
                findWord = []
            findWord.append(currentNode.char)
        if(currentNode.value is not None):
            wordAndKey = [''.join(findWord), currentNode.value]
            writeCSVfile('exit.csv', wordAndKey)    # writes word and key in csv file
            findWord = None
        for son in currentNode.sons:
            getAllTrie(son, findWord)

root = Node()
testStr = 'querida queremos ate quero quero'
listStr = testStr.split( )
for word in listStr:
    root = insertTrie(root, word, 0)

#print(root) -> print root outside of function, 'q' doesn't appear
getAllTrie(root, None)

当我在函数“ getAllTrie”（在代码的注释中）之外打印“ root”时，它打印以下内容：

<__main__.Node object at 0x03323610>

但是，当我将其打印在函数内部（也在注释中）时，它会打印以下内容：

<__main__.Node object at 0x03323610>
q

我不知道为什么会有“ q”。 它是根的一个儿子中包含的字符，但是当我在函数中打印根本身时，它就会显示出来，我也不知道为什么。

Answer 1

您的sons属性是一个dict，将单字符字符串映射到节点。

因此，当您这样做时：

for son in currentNode.sons:

…每个son都是一个单字符str对象，而不是Node对象。 这就是为什么第一个递归调用会打印出q原因，并且为什么它会引发关于没有sons属性的'q'字符串的异常。

如果要遍历字典中的值而不是键 ，则需要这样说：

for son in currentNode.sons.values():

---

您的代码中还有其他多个错误，从无效的缩进到ccurrentChar拼写错误，而且它是不完整的（任何地方都没有writeCSVFile函数），因此我无法测试这是否真的可以修复您的函数，但是可以解决此特定错误。