Trie(发音类似 "try")或者说 前缀树 是一种树形数据结构,用于高效地存储和检索字符串数据集中的键。这一数据结构有相当多的应用情景,例如自动补全和拼写检查。
请你实现 Trie 类:
Trie()
初始化前缀树对象。void insert(String word)
向前缀树中插入字符串word
。boolean search(String word)
如果字符串word
在前缀树中,返回true
(即,在检索之前已经插入);否则,返回false
。boolean startsWith(String prefix)
如果之前已经插入的字符串word
的前缀之一为prefix
,返回true
;否则,返回false
。
示例:
输入 ["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
提示:
1 <= word.length, prefix.length <= 2000
word
和prefix
仅由小写英文字母组成insert
、search
和startsWith
调用次数 总计 不超过3 * 104
次
前缀树是一个很实用的东西,现实中有很多用的地方,做算法题也经常会用到
这个是相对简单的前缀树的版本,正常的前缀树的每个节点还应该包含经过的次数的信息,这个如果后面解答其他题的时候需要我就贴上来
其他的就不多说了,上代码,看不懂的请留言或者私信,收到第一时间解答
class Trie {
/**定义前缀树的根,这个根没有任何意义,只是查找和插入的起点 */
TrieNode root;
public Trie() {
root = new TrieNode();
}
public void insert(String word) {
TrieNode cur = root;
/**转成字符数组 */
char[] wordArr = word.toCharArray();
for(char c : wordArr) {
/**如果路径没有先创建 */
if(cur.nexts[c-'a'] == null) {
cur.nexts[c-'a'] = new TrieNode();
}
cur = cur.nexts[c-'a'];
}
/**最后cur是我们遍历的最后一个节点,我们要把它的end属性设置为true */
cur.isEnd = true;
}
public boolean search(String word) {
/**转成字符数组操作*/
char[] wordArr = word.toCharArray();
TrieNode cur = root;
for(char c : wordArr) {
/**路径不存在直接返回false */
if(cur.nexts[c-'a'] == null) {
return false;
}
/**路径存在接着找 */
cur = cur.nexts[c-'a'];
}
return cur.isEnd;
}
public boolean startsWith(String prefix) {
/**转成字符数组操作*/
char[] wordArr = prefix.toCharArray();
TrieNode cur = root;
for(char c : wordArr) {
/**路径不存在直接返回false */
if(cur.nexts[c-'a'] == null) {
return false;
}
/**路径存在接着找 */
cur = cur.nexts[c-'a'];
}
/**这个方法看的是不是以这个前缀开头,现在前缀我们走完了,当然是true了 */
return true;
}
static class TrieNode {
/**从26条路径分别代表a~z的往下的路径*/
TrieNode[] nexts;
/**是否是结尾 */
boolean isEnd;
public TrieNode() {
nexts = new TrieNode[26];
}
}
}
/**
* Your Trie object will be instantiated and called as such:
* Trie obj = new Trie();
* obj.insert(word);
* boolean param_2 = obj.search(word);
* boolean param_3 = obj.startsWith(prefix);
*/