//Implement Trie (Prefix Tree)
//Implement a trie with insert, search, and startsWith methods.
//You may assume that all inputs are consist of lowercase letters a-z.
class MYTrieNode {
char val;
boolean isWord;
MYTrieNode[] subNodes = new MYTrieNode[26];
MYTrieNode(char val) {
this.val = val;
}
}
class Trie {
private MYTrieNode root;
/** Initialize your data structure here. */
public Trie() {
root = new MYTrieNode(' ');
}
/** Inserts a word into the trie. */
public void insert(String word) {
MYTrieNode node = root;
for (int i = 0; i < word.length(); i++) {
if (node.subNodes[word.charAt(i) - 'a'] != null) {
node = node.subNodes[word.charAt(i) - 'a'] ;
} else {
node.subNodes[word.charAt(i) - 'a'] = new MYTrieNode(word.charAt(i));
node = node.subNodes[word.charAt(i) - 'a'];
}
}
node.isWord = true;
}
/** Returns if the word is in the trie. */
public boolean search(String word) {
MYTrieNode node = root;
for (int i = 0; i < word.length(); i++) {
if (node.subNodes[word.charAt(i) - 'a'] != null) {
node = node.subNodes[word.charAt(i) - 'a'] ;
} else {
return false;
}
}
return node.isWord;
}
/** Returns if there is any word in the trie that starts with the given prefix. */
public boolean startsWith(String prefix) {
MYTrieNode node = root;
for (int i = 0; i < prefix.length(); i++) {
if (node.subNodes[prefix.charAt(i) - 'a'] != null) {
node = node.subNodes[prefix.charAt(i) - 'a'] ;
} else {
return false;
}
}
return true;
}
}
实现前缀树(增、判断是否有该单词,是否有该前缀)
最后编辑于 :
©著作权归作者所有,转载或内容合作请联系作者
- 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
- 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
- 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
推荐阅读更多精彩内容
- 成长记录-连载(三十六) ——我的第一篇五千字长文,说了什么,你一定想不到 并不是不想每天写公众号,而是之前思考怎...
- 20+个很棒的Android开源项目本文摘自文章: 20+ Awesome Open-Source Android...