LeetCode No648. 单词替换

题目

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

现在，给定一个由许多词根组成的词典 dictionary 和一个用空格分隔单词形成的句子 sentence。你需要将句子中的所有继承词用词根替换掉。如果继承词有许多可以形成它的词根，则用最短的词根替换它。

你需要输出替换之后的句子。

示例 1：

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

示例 2：

输入：dictionary = ["a","b","c"], sentence = "aadsfasf absbs bbab cadsfafs"
输出："a a b c"

提示：

1 <= dictionary.length <= 1000
1 <= dictionary[i].length <= 100
dictionary[i] 仅由小写字母组成。
1 <= sentence.length <= 10^6
sentence 仅由小写字母和空格组成。
sentence 中单词的总量在范围 [1, 1000] 内。
sentence 中每个单词的长度在范围 [1, 1000] 内。
sentence 中单词之间由一个空格隔开。
sentence 没有前导或尾随空格。

思路

一个字符前缀的题，标准的字典树模板题，对于学习字典树模板还是比较直观的。

AC代码

点击查看代码

class Solution {
    class Node {
        boolean isEnd;
        Node[] nodes = new Node[26];
    }

    Node root = new Node();

    private void add(String s) {
        Node head = root;
        for(int i=0; i<s.length(); i++) {
            int chNum = (int)(s.charAt(i)-'a');
            if( head.nodes[chNum] == null ) {
                head.nodes[chNum] = new Node();
            }
            head = head.nodes[chNum];
        }
        head.isEnd = true;
    }

    private String query(String s) {
        Node head = root;
        for(int i=0; i<s.length(); i++) {
            int chNum = (int)(s.charAt(i)-'a');
            if( head.nodes[chNum]==null ) {
                break;
            }
            if( head.nodes[chNum].isEnd ) {
                return s.substring(0, i+1);
            }
            head = head.nodes[chNum];
        }
        return s;
    }

    public String replaceWords(List<String> dictionary, String sentence) {
        for(String str: dictionary) {
            add(str);
        }
        StringBuilder sb = new StringBuilder();
        for (String str : sentence.split(" ")) {
            sb.append(query(str)).append(" ");
        }
        return sb.substring(0, sb.length() - 1);
    }
}