一、题目
二、示例
输入
["Trie", "insert", "search", "search", "startsWith", "insert", "search"]
[[], ["apple"], ["apple"], ["app"], ["app"], ["app"], ["app"]]
输出
[null, null, true, false, true, null, true]
解释
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
三、思路
如上图所示,可以实现树,当插入单词的时候,遍历每一个字母,如果该字母存在,则继续向下遍历,如果 不存在就以该字母的为key赋值为一个对象,当插入完全后,则给该对象设置一个值为isEnd = true
。
四、代码
var Trie = function () {
this.root = {}
};
/**
* @param {string} word
* @return {void}
*/
Trie.prototype.insert = function (word) {
let node = this.root
for (let item of word) {
if (!node[item]) {
node[item] = {}
}
node = node[item]
}
node.isEnd = true
};
/**
* @param {string} word
* @return {boolean}
*/
Trie.prototype.search = function (word) {
let node = this.root
for (let item of word) {
if (!node[item]) {
return false
}
node = node[item]
}
return node.isEnd ? true : false
};
/**
* @param {string} prefix
* @return {boolean}
*/
Trie.prototype.startsWith = function (prefix) {
let node = this.root
for(let item of prefix) {
if(!node[item]) {
return false
}
node = node[item]
}
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)
*/
五、总结