java 前缀树实现 leetcode 208: Implement Trie (Prefix Tree)

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;
    }

}