用 JavaScript 实现单词查找树

2024-02-07 16:32:37

Trie 树简介

Trie 树（又名单词查找树）是一种树形数据结构，用于存储字符串。Trie 树的每个节点代表一个字符，并且每个节点的子节点代表该字符的下一个可能字符。通过这种方式，Trie 树可以存储大量字符串，并高效地搜索这些字符串。

Trie 树具有以下几个优点：

搜索字符串的时间复杂度为 O(m)，其中 m 是字符串的长度。
可以高效地进行前缀搜索，即搜索所有以给定前缀开头的字符串。
可以高效地插入和删除字符串。
可以用于实现自动完成、拼写检查和数据压缩等应用。

JavaScript 实现 Trie 树

下面是一个用 JavaScript 实现 Trie 树的示例代码：

class TrieNode {
  constructor() {
    this.children = {};
    this.isEndOfWord = false;
  }
}

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

  insert(word) {
    let currentNode = this.root;
    for (let i = 0; i < word.length; i++) {
      const char = word[i];
      if (!currentNode.children[char]) {
        currentNode.children[char] = new TrieNode();
      }
      currentNode = currentNode.children[char];
    }
    currentNode.isEndOfWord = true;
  }

  search(word) {
    let currentNode = this.root;
    for (let i = 0; i < word.length; i++) {
      const char = word[i];
      if (!currentNode.children[char]) {
        return false;
      }
      currentNode = currentNode.children[char];
    }
    return currentNode.isEndOfWord;
  }

  startsWith(prefix) {
    let currentNode = this.root;
    for (let i = 0; i < prefix.length; i++) {
      const char = prefix[i];
      if (!currentNode.children[char]) {
        return false;
      }
      currentNode = currentNode.children[char];
    }
    return true;
  }
}