用 JavaScript 实现单词查找树

作者：Hussain Mir Ali

翻译：疯狂的技术宅

原文：https://www.softnami.com/post...

未经允许严禁转载

动机

对于搜索字符串的需求，在最坏的情况下，二叉搜索树的时间复杂度可能为 O(n)，“n” 是二叉树中存储的字符串的总数量。所以为了在最佳时间内搜索字符串，需要一种性能更好的数据结构。 Trie 树（又名单词搜索树）可以避免在搜索字符串时遍历整个树。仅包含字母的字符串会把 trie 节点的子级数量限制为 26。这样搜索字符串的时间复杂度为 O(s)，其中 “s” 为字符串的长度。与二进制搜索树相比，trie 树在搜索字符串方面效率更高。

方法

trie 树中单个节点的结构由长度为 26 的数组和一个布尔值组成，这个布尔值用来标识其是否为叶子节点。此外，叶子节点可以具有整数值或映射到字符串的其他类型的值。数组中的每个索引代表从 a 到 z 的字母，并且每个索引可以有一个 TrieNode 实例。

上图表示 trie 树中的根节点。

实现

该实现包含两个类，一个用于 trie 节点，另一个用于 trie 树。实现的语言是带有 ES6 规范的 JavaScript。

TrieNode 类的属性为value，isEnd和 arr。变量 arr 是长度为 26 的数组，其中填充了 null 值。

class TrieNode {
    constructor() {
        this.value = undefined;
        this.isEnd = false;
        this.arr = new Array(26).fill(null);
    }
}

TrieTree 类具有 insert、searchNode、startsWith、printString 和 getRoot 方法。可以用 startsWith 方法执行前缀搜索。insert方法将字符串和值作为参数。

class TrieTree {

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

    insert(word, value) {
        let node = this.root;
        for (let i = 0; i < word.length; i++) {
            const index = parseInt(word[i], 36) - 10;

            if (node.arr[index] === null) {
                const temp = new TrieNode();
                node.arr[index] = temp;
                node = temp;
            } else {
                node = node.arr[index];
            }
        }
        node.isEnd = true;
        node.value = value;
    }

    getRoot() {
        return this.root;
    }

    startsWith(prefix) {
        const node = this.searchNode(prefix);
        if (node == null) {
            return false;
        } else {
            this.printStrings(node, "");
            return true;
        }
    }

    printStrings(node, prefix) {
        if (node.isEnd) console.log(prefix);
        for (let i = 0; i < node.arr.length; i++) {
            if (node.arr[i] !== null) {
                const character = String.fromCharCode('a'.charCodeAt() + i);
                this.printStrings(node.arr[i], prefix + "" + (character));
            }
        }
    }

    searchNode(str) {
        let node = this.root;
        for (let i = 0; i < str.length; i++) {
            const index = parseInt(str[i], 36) - 10;
            if (node.arr[index] !== null) {
                node = node.arr[index];
            } else {
                return null;
            }
        }

        if (node === this.root)
            return null;

        return node;
    }
}

下面的代码显示了如何实例化 “TrieTree” 类，还演示了各种方法的用法。

const trieTree = new TrieTree();
trieTree.insert("asdfasdf", 5);
trieTree.insert("cdfasdfas", 23);
trieTree.insert("cdfzsvljsdf", 42);

let answer = trieTree.searchNode("asdfasdf");
console.log(answer.value); //5
answer = trieTree.startsWith("cdf");
console.log(answer);
//asdfas
//zsvljsdf
//true

不同方法的时间和空间复杂度如下：

• searchNode：时间复杂度：O(s)，'s' 是字符串的长度，空间复杂度：O(1)，因为没有使用额外的数据结构，例如队列或栈。
• insert：时间复杂度：O(s)，“s”是字符串长度，空间复杂度：O(ns)，其中 “n” 是 trie 树中 key 的数量，“s” 是字符串的长度。
• startsWith：时间复杂度：O(s)，'s' 是字符串的长度，'k' 是剩余匹配字符串的最大长度，空间复杂度：O(k)，其中 'k' 是其余匹配字符串的最大长度。

应用

在前端开发中，trie 树可用于以下程序：

• 自动完成和提前输入功能。
• 拼写检查。
• 搜索。
• 排序。

此外 trie 树可以用来存储电话号码、IP地址和对象等。

---

本文首发微信公众号：前端先锋

欢迎扫描二维码关注公众号，每天都给你推送新鲜的前端技术文章

欢迎继续阅读本专栏其它高赞文章：

• 深入理解Shadow DOM v1
• 一步步教你用 WebVR 实现虚拟现实游戏
• 13个帮你提高开发效率的现代CSS框架
• 快速上手BootstrapVue
• JavaScript引擎是如何工作的？从调用栈到Promise你需要知道的一切
• WebSocket实战：在 Node 和 React 之间进行实时通信
• 关于 Git 的 20 个面试题
• 深入解析 Node.js 的 console.log
• Node.js 究竟是什么？
• 30分钟用Node.js构建一个API服务器
• Javascript的对象拷贝
• 程序员30岁前月薪达不到30K，该何去何从
• 14个最好的 JavaScript 数据可视化库
• 8 个给前端的顶级 VS Code 扩展插件
• Node.js 多线程完全指南
• 把HTML转成PDF的4个方案及实现

---

• 更多文章...