LeetCode 648中等Trie
单词替换 图解题解
这道题到底在问什么
给定一个由许多 词根 组成的词典 dictionary 和一个用空格分隔单词形成的句子 sentence 。你需要将句子中的所有 衍生词 用 词根 替换掉。如果 衍生词 有许多可以形成它的 词根 ，则用 最短 的 词根 替换它。 你需要输出替换之后的句子。
- dictionary,sentence
- ["cat","bat","rat"], "the cattle was rattled by the battery"
- 输出
- "the cat was rat by the bat"
先想最直接的笨办法
数组/字符串状态条跟着代码走：推进语句是：return ' '.join(ans)。处理过的部分不再重新枚举。（动画第 10 步）
最优解：一步一步想明白
- 3下面 9 步动画会按主解代码推进，而不是泛泛讲题型。
- 4先读清 单词替换 的输入输出数组/字符串状态条跟着代码走：先把示例输入映射到代码参数：def replaceWords(self, dictionary, sentence):。
- 5roots = set(dictionary)；ans = []数组/字符串状态条跟着代码走：开局只立住必要变量：roots = set(dictionary)；ans = []。
- 6for word in sentence.split():数组/字符串状态条跟着代码走：主流程从这里开始：for word in sentence.split():。
- 7if prefix in roots:数组/字符串状态条跟着代码走：题目条件落到这一行：if prefix in roots:。
- 8prefix += ch数组/字符串状态条跟着代码走：对应代码：prefix += ch。这一行决定当前轮对答案有什么贡献。
- 9找最短词根数组/字符串状态条跟着代码走：边界跟着代码看：ans.append(replaced)。
- 10return ' '.join(ans)数组/字符串状态条跟着代码走：推进语句是：return ' '.join(ans)。处理过的部分不再重新枚举。
- 11return ' '.join(ans)数组/字符串状态条跟着代码走：到这里，ans 已经能表达题目要求。
- 12return：return ' '.join(ans)数组/字符串状态条跟着代码走：最后检查返回形态：返回值、原地修改或设计类状态，要和 LeetCode 判题方式一致。
- 15记住这题的代码骨架：题意约束先落到状态变量，再用循环或递归维护它。
⚠️ 容易写错的地方
✗ 错：找最长词根
✓ 对：找最短词根
题目要求最短可替换前缀
✗ 错：替换后继续查更长
✓ 对：找到立刻停止
最短词根优先
完整代码（Python）
Python
class Solution:
def replaceWords(self, dictionary, sentence):
roots = set(dictionary)
ans = []
for word in sentence.split():
prefix = ''
replaced = word
for ch in word:
prefix += ch
if prefix in roots:
replaced = prefix
break
ans.append(replaced)
return ' '.join(ans)复杂度
时间复杂度
O(total chars)
扫描所有单词前缀
空间复杂度
O(dictionary chars)
词根集合
看不够？换成动画再走一遍
上面的推演每一步都对应一帧动画。点开交互动画版，能一步步看着 单词替换 的数据怎么变、指针怎么走，还能切 Python / Java / C++ 跟着练。
面试官可能追问
这道题为什么用「Trie」，换最直接的暴力解会差在哪？+
Trie抓住了本题的结构特征，把暴力解里重复的工作省掉；暴力解通常要多嵌套一层枚举，数据一大就超时。具体对比见上文「暴力解及其卡点」与「最优解逐步推演」两节。
时间复杂度为什么是 undefined？怎么推出来的？+
按上文复杂度小节的推导，时间复杂度为 undefined。
本平台为独立第三方培训机构，与华为技术有限公司无任何关联；课程的服务内容与权益以购买协议为准，学习效果因个人情况而异。「华为 OD」「华为可信」等仅为对岗位与考试方向的客观描述，相关商标归各自权利人所有。