在 Java 中实现基本的自动完成

Question

给定一个单词和短语数组，给定一个键，返回：

• 所有以key开头的单词
• 包含以 key 开头的单词的所有短语

例如：

wordbank = ["bang", "base", "bore", "band", "This is a baffling problem];

key = "ba";

autocomplete(wordbank, key) 应该返回 ["bang", "base", "band", "This is a baffling problem]

我用 Trie 来做，但想知道 this is a good solution?

要运行，只需在终端中输入java Test。链接里的测试用例和这里的例子不一样。

Answer 1

代码可以通过以下方式简化。向 try 的每个节点添加一组自动完成结果。在尝试插入单词时，同时为每个访问的节点添加自动完成集。要执行自动完成，只需返回相应节点的自动完成集即可。

这个解决方案在insertWord和getWordsWithPrefix上增加了一行，同时完全去掉了buildWords的需要。