前缀树-单词替换

648. 单词替换

在英语中，我们有一个叫做 词根(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 没有前导或尾随空格。

/**
 * @param {string[]} dictionary
 * @param {string} sentence
 * @return {string}
 */
var replaceWords = function(dictionary, sentence) {
  const trie = new Trie();
  for (const word of dictionary) {
    trie.insert(word);
  }

  const sentenceArr = sentence.split(' ');
  for (let i = 0; i < sentenceArr.length; i++) {
    sentenceArr[i] = trie.searchRoot(sentenceArr[i]);
  }

  return sentenceArr.join(' ');
};

class Trie {
  constructor() {
    this.root = {};
  }

  insert(word) {
    let curNode = this.root;
    for (const ch of word) {
      if (!curNode[ch]) {
        curNode[ch] = {};
      }
      curNode = curNode[ch];
    }

    curNode.end = true;
  }

  searchRoot(word) {
    let curNode = this.root;
    let result = '';
    for (const ch of word) {
      if (!curNode[ch]) {
        return word;
      } else {
        result += ch;
        if (curNode[ch].end) {
          return result;
        }
      }
      curNode = curNode[ch];
    }

    return word;
  }
}