我编写了一个程序来在三元搜索树中查找所有可能的单词以及给定字母的列表。输出是所有单词的排序集。以下是查找单词的python代码:
def _find_words(self, node: Node, letters: str, buffer=''):
if node is None:
return
for c in letters:
if node.key < c:
self._find_words(node.high, letters, '')
elif node.key > c:
self._find_words(node.low, letters, '')
else:
buffer += node.key
if node.isEnd:
self.words.add(buffer)
self._find_words(node.equal,
letters.replace(node.key, '', 1), buffer)
树包含以下单词:cat、acts、act、actor、at、bug、cats、up、actors、upper
我得到以下字母“abugpscort”的输出:
['acts', 'cats', 'cat', 'act', 'bug', 'ors', 'up', 'or', 't']
而输出应该是:
['actors', 'actor', 'acts', 'cats', 'cat', 'act', 'bug', 'up', 'at']
我怎样才能解决这个问题?