算法题单词替换

最新推荐文章于 2025-12-05 14:37:41 发布

原创

最新推荐文章于 2025-12-05 14:37:41 发布 · 660 阅读

19 ·

CC 4.0 BY-SA版权

文章标签：

#算法

648. 单词替换

问题描述

在英语中，我们有一个叫做"词根"（root）的概念，可以词根后面添加其他一些词组成另一个较长的单词——我们称这个词为"继承词"（successor）。例如，词根"an"跟随着单词"other"，可以形成新的单词"another"。

现在，给定一个由许多词根组成的词典和一个句子，需要将句子中的每个继承词用其词根替换。如果继承词有许多可以形成它的词根，则用最短的词根替换它。

返回替换之后的句子。

示例：

输入: dictionary = ["cat","bat","rat"], sentence = "the cattle was rattled by the battery"
输出: "the cat was rat by the bat"

算法思路

核心思想：前缀树（Trie） + 字符串处理

前缀树构建：将词典中的所有词根构建成前缀树
单词替换：对句子中的每个单词，在前缀树中查找最短词根
字符串处理：按空格分割句子，处理每个单词

优化：

前缀树：高效查找前缀，避免逐个比较所有词根
最短词根：在前缀树中首次找到的词根就是最短的

代码实现

import java.util.*;

class Solution {
   
   
    /**
     * 将句子中的继承词替换为其最短词根
     * 
     * @param dictionary 词根词典
     * @param sentence 需要处理的句子
     * @return 替换后的句子
     */
    public String replaceWords(List<String> dictionary, String sentence) {
   
   
        // 构建前缀树
        Trie trie = new Trie();
        
        // 将所有词根插入前缀树
        for (String root : dictionary) {
   
   
            trie.insert(root);
        }
        
        // 按空格分割句子
        String[] words = sentence.split("\\s+");
        StringBuilder result = new StringBuilder();
        
        // 处理每个单词
        for (int i = 0; i < words.length; i++) {
   
   
            if (i > 0) {
   
   
                result.append(" ");
            }
            
            String word = words[i];
            String root = trie.findShortestRoot(word);
            
            // 如果找到词根，使用词根；否则使用原单词
            result.append