中等

问题描述

在英语中，我们有一个叫做 词根(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 <= 106
• sentence 仅由小写字母和空格组成。
• sentence 中单词的总量在范围 [1, 1000] 内。
• sentence 中每个单词的长度在范围 [1, 1000] 内。
• sentence 中单词之间由一个空格隔开。
• sentence 没有前导或尾随空格。

排序

先将 $dictionary$ 中的词根按长度排序，然后遍历 $sentence$ 中的单词，从排序后的 $dictionary$ 中查找是否存在某个单词是当前句子中单词的词根，若是，则替换掉当前单词。

/**
 * @param {string[]} dictionary
 * @param {string} sentence
 * @return {string}
 */
var replaceWords = function (dictionary, sentence) {
  dictionary.sort((a, b) => a.length - b.length)
  return sentence
    .split(' ')
    .map((word) => {
      for (const string of dictionary) {
        if (word.startsWith(string)) return string
      }
      return word
    })
    .join(' ')
}

• 时间复杂度：$O(n \times m)$，$m$ 为词根的数量，$n$ 为单词的数量
• 空间复杂度：$O(n)$

字典树

使用 $dictionary$ 构建字典树，在遍历 $sentence$ 中的单词，从字典树中查找替换。

/**
 * @param {string[]} dictionary
 * @param {string} sentence
 * @return {string}
 */
var replaceWords = function (dictionary, sentence) {
  const root = new Map()
  const add = (word) => {
    let node = root
    for (let i = 0; i < word.length; i++) {
      if (!node.has(word[i])) {
        node.set(word[i], new Map())
      }
      node = node.get(word[i])
    }
    node.set('#', true)
  }
  const search = (word) => {
    let node = root
    for (let i = 0; i < word.length; i++) {
      if (!node.has(word[i])) {
        return null
      }
      node = node.get(word[i])
      if (node.get('#')) return word.substr(0, i + 1)
    }
    return null
  }
  dictionary.forEach((word) => add(word))
  return sentence
    .split(' ')
    .map((word) => search(word) || word)
    .join(' ')
}

• 时间复杂度：$O(d + s)$，$d$ 是 $dictionary$ 的字符数，$s$ 是 $sentence$ 的字符数。
• 空间复杂度：$O(d + s)$