Implement a trie with insert
, search
, and startsWith
methods.
1.前缀树的概念
所有含有公共前缀的字符串将挂在树中同一个结点下。实际上trie
简明的存储了存在于串集合中的所有公共前缀。
2.举例说明
假如有这样一个字符串集合
X{bear,bell,bid,bull,buy,sell,stock,stop}
。它的标准
Trie
树如下图:
生成的树如图所示。我们可以在每个节点设置item用于存储一个单词。
3.性质
根节点不包含字符,除根节点外每一个节点都只包含一个字符;
从根节点到某一节点,路径上经过的字符连接起来,为该节点对应的字符串;
每个节点的所有子节点包含的字符都不相同。
4.查找
标准 Trie树的查找:
对于英文单词的查找,我们完全可以在内部结点中建立26个元素组成的指针数组。如果要查找a,只需要在内部节点的指针数组中找第0个指针即可(b=第1个指针,随机定位)。时间复杂度为O(1)。
查找过程:假如我们要在上面那棵Trie中查找字符串bull (b-u-l-l)。
(1) 在root结点中查找第(‘b’-‘a’=1)号孩子指针,发现该指针不为空,则定位到第1号孩子结点处——b结点。
(2) 在b结点中查找第(‘u’-‘a’=20)号孩子指针,发现该指针不为空,则定位到第20号孩子结点处——u结点。
(3) … 一直查找到叶子结点出现特殊字符‘$’位置,表示找到了bull字符串
如果在查找过程中终止于内部结点,则表示没有找到待查找字符串。
效率:
对于有n个英文字母的串来说,在内部结点中定位指针所需要花费O(d)时间,d为字母表的大小,英文为26。由于在上面的算法中内部结点指针定位使用了数组随机存储方式,因此时间复杂度降为了O(1)。但是如果是中文字,下面在实际应用中会提到。因此我们在这里还是用O(d)。 查找成功的时候恰好走了一条从根结点到叶子结点的路径。因此时间复杂度为O(d*n)。
但是,当查找集合X中所有字符串两两都不共享前缀时,trie中出现最坏情况。除根之外,所有内部结点都自由一个子结点。此时的查找时间复杂度蜕化为O(d*(n^2)
5.JAVA源代码
class TrieNode {
//存储子节点
TrieNode[] children=new TrieNode[26];
//存储一条记录,即对应一个单词
String item="";
// Initialize your data structure here.
public TrieNode() {
}
}
public class Trie {
private TrieNode root;
public Trie() {
root = new TrieNode();
}
// Inserts a word into the trie.
public void insert(String word) {
TrieNode node=this.root;
for(char c:word.toCharArray())
{
if(node.children[c-'a']==null)
{
node.children[c-'a']=new TrieNode();
}
node=node.children[c-'a'];
}
node.item=word;
}
// Returns if the word is in the trie.
public boolean search(String word) {
TrieNode node=this.root;
for(char c:word.toCharArray())
{
if(node.children[c-'a']==null)
{
return false;
}
node=node.children[c-'a'];
}
return node.item.equals(word);
}
// Returns if there is any word in the trie
// that starts with the given prefix.
public boolean startsWith(String prefix) {
TrieNode node=this.root;
for(char c:prefix.toCharArray())
{
if(node.children[c-'a']==null)
{
return false;
}
node=node.children[c-'a'];
}
return true;
}
}