[100天算法】-实现 Trie(day 41)

题目描述

实现一个 Trie (前缀树),包含 insert, search, 和 startsWith 这三个操作。

示例:

Trie trie = new Trie();

trie.insert("apple");
trie.search("apple"); // 返回 true
trie.search("app"); // 返回 false
trie.startsWith("app"); // 返回 true
trie.insert("app");
trie.search("app"); // 返回 true
说明:

你可以假设所有的输入都是由小写字母 a-z 构成的。
保证所有输入均为非空字符串。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/implement-trie-prefix-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

思路

熟悉一下 Trie 的概念就可以了。

https://medium.com/basecs/trying-to-understand-tries-3ec6bede0014

复杂度分析

  • 时间复杂度:O(L),L 是字符串长度, insert search startsWith 操作都是。
  • 空间复杂度:O(M\^{L}),L 是字符串长度,M 是字符集中字符个数,如本题中 M 就是 26。

代码

TypeScript Code

复制代码
class TrieNode {
    value: string;
    children: Array<TrieNode | null>;

    constructor(value) {
        this.value = value;
        this.children = Array(26);
    }
}

class Trie {
    private root: TrieNode;

    constructor() {
        this.root = this._getTrieNode('');
    }

    private _getTrieNode(value: string): TrieNode {
        return new TrieNode(value);
    }

    private _char2Index(char: string): number {
        return char.toLowerCase().charCodeAt(0) - 97;
    }

    insert(word: string): void {
        let crawl: TrieNode = this.root;
        for (let char of word) {
            const index: number = this._char2Index(char);
            if (!crawl.children[index]) {
                crawl.children[index] = this._getTrieNode('');
            }
            crawl = crawl.children[index];
        }
        crawl.value = word;
    }

    search(word: string): boolean {
        let crawl: TrieNode = this.root;
        for (let char of word) {
            const index: number = this._char2Index(char);
            if (!crawl.children[index]) return false;
            crawl = crawl.children[index];
        }
        return crawl.value === word;
    }

    startsWith(prefix: string): boolean {
        let crawl: TrieNode = this.root;
        for (let char of prefix) {
            const index: number = this._char2Index(char);
            if (!crawl.children[index]) return false;
            crawl = crawl.children[index];
        }
        return true;
    }
}

/**
 * Your Trie object will be instantiated and called as such:
 * var obj = new Trie()
 * obj.insert(word)
 * var param_2 = obj.search(word)
 * var param_3 = obj.startsWith(prefix)
 */

JavaScript Code

复制代码
class TrieNode {
    constructor(val) {
        this.value = val;
        this.pointers = Array(26);
    }
}

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

    /**
     * @param {string} val
     */
    _getTrieNode(val) {
        return new TrieNode(val);
    }

    /**
     * @param {string} char
     * @returns {number}
     */
    _char2Index(char) {
        return char.toLowerCase().charCodeAt(0) - 97;
    }

    /**
     * Inserts a word into the trie.
     * @param {string} word
     * @return {void}
     */
    insert(word) {
        let crawl = this.root;
        for (let char of word) {
            const index = this._char2Index(char);
            if (!crawl.pointers[index]) {
                crawl.pointers[index] = this._getTrieNode('');
            }
            crawl = crawl.pointers[index];
        }
        // Store the word in the last TrieNode as an end mark.
        crawl.value = word;
    }

    /**
     * Returns if the word is in the trie.
     * @param {string} word
     * @return {boolean}
     */
    search(word) {
        let crawl = this.root;
        for (let char of word) {
            const index = this._char2Index(char);
            if (!crawl.pointers[index]) return false;

            crawl = crawl.pointers[index];
        }
        // If it has a stored value, it is the last TrieNode, i.e., the desired word is found.
        // Otherwise, the word doesn't exist in Trie.
        return !!crawl.value;
    }

    /**
     * Returns if there is any word in the trie that starts with the given prefix.
     * @param {string} prefix
     * @return {boolean}
     */
    startsWith(prefix) {
        let crawl = this.root;
        for (let char of prefix) {
            const index = this._char2Index(char);
            if (!crawl.pointers[index]) return false;

            crawl = crawl.pointers[index];
        }
        return true;
    }
}

输入输出

Nodejs

复制代码
const __main__ = function () {
    const readline = require('readline');
    const rl = readline.createInterface({
        input: process.stdin,
        output: process.stdout,
    });

    console.log('******输入******');
    rl.prompt();
    const lines = [];
    rl.on('line', line => lines.push(line));

    rl.on('close', () => {
        console.log('\n******输出******');

        const test = (operations, params) => {
            if (operations[0] === 'Trie') {
                const trie = new Trie();
                const output = [null];
                for (let i = 1; i < operations.length; i++) {
                    const res = trie[operations[i]](...params[i]);
                    output.push(res === void 0 ? null : res);
                }
                console.log(output);
            } else {
                console.log(Array(operations.length).fill(null));
            }
        };

        while (lines.length >= 2) {
            const params = lines.splice(0, 2);
            test(...params.map(el => JSON.parse(el)));
        }
    });
};
相关推荐
为什么这亚子42 分钟前
九、Go语言快速入门之map
运维·开发语言·后端·算法·云原生·golang·云计算
1 小时前
开源竞争-数据驱动成长-11/05-大专生的思考
人工智能·笔记·学习·算法·机器学习
~yY…s<#>1 小时前
【刷题17】最小栈、栈的压入弹出、逆波兰表达式
c语言·数据结构·c++·算法·leetcode
幸运超级加倍~2 小时前
软件设计师-上午题-16 算法(4-5分)
笔记·算法
yannan201903132 小时前
【算法】(Python)动态规划
python·算法·动态规划
埃菲尔铁塔_CV算法2 小时前
人工智能图像算法:开启视觉新时代的钥匙
人工智能·算法
EasyCVR2 小时前
EHOME视频平台EasyCVR视频融合平台使用OBS进行RTMP推流,WebRTC播放出现抖动、卡顿如何解决?
人工智能·算法·ffmpeg·音视频·webrtc·监控视频接入
linsa_pursuer2 小时前
快乐数算法
算法·leetcode·职场和发展
小芒果_012 小时前
P11229 [CSP-J 2024] 小木棍
c++·算法·信息学奥赛
qq_434085902 小时前
Day 52 || 739. 每日温度 、 496.下一个更大元素 I 、503.下一个更大元素II
算法