首页 > 代码库 > Trie树(转)

Trie树(转)

Trie 树, 又称字典树,单词查找树。它来源于retrieval(检索)中取中间四个字符构成(读音同try)。用于存储大量的字符串以便支持快速模式匹配。主要应用在信息检索领域。

 

Trie 有三种结构: 标准trie (standard trie)、压缩trie、后缀trie(suffix trie) 。 最后一种将在《字符串处理4:后缀树》中详细讲,这里只将前两种。

 

 

1. 标准Trie (standard trie)

 

标准 Trie树的结构 : 所有含有公共前缀的字符串将挂在树中同一个结点下。实际上trie简明的存储了存在于串集合中的所有公共前缀。 假如有这样一个字符串集合X{bear,bell,bid,bull,buy,sell,stock,stop}。它的标准Trie树如下图:

 

 

      上图(蓝色圆形结点为内部结点,红色方形结点为外部结点),我们可以很清楚的看到字符串集合X构造的Trie树结构。其中从根结点到红色方框叶子节点所经历的所有字符组成的串就是字符串集合X中的一个串。

 

      注意这里有一个问题: 如果X集合中有一个串是另一个串的前缀呢? 比如,X集合中加入串bi。那么上图的Trie树在绿色箭头所指的内部结点i 就应该也标记成红色方形结点。这样话,一棵树的枝干上将出现两个连续的叶子结点(这是不合常理的)。

 

      也就是说字符串集合X中不存在一个串是另外一个串的前缀 。如何满足这个要求呢?我们可以在X中的每个串后面加入一个特殊字符$(这个字符将不会出现在字母表中)。这样,集合X{bear$、bell$、.... bi$、bid$}一定会满足这个要求。

 

      总结:一个存储长度为n,来自大小为d的字母表中s个串的集合X的标准trie具有性质如下:

      (1) 树中每个内部结点至多有d个子结点。

      (2) 树有s个外部结点。

      (3) 树的高度等于X中最长串的长度。

      (4) 树中的结点数为O(n)。

 

 

标准 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))

       

 

标准 Trie树的Java代码实现:

 

[java] view plaincopy
 
  1. /* 
  2. StandarTire.java 
  3. Trie 树, 又称字典树,单词查找树。 
  4. 它来源于retrieval(检索)中取中间四个字符构成(读音同try)。用于存储大量的字符串以便支持快速模式匹配。主要应用在信息检索领域。 
  5.  
  6. @author arhaiyun 
  7. date:2013/09/23 
  8. */  
  9.   
  10. import java.util.*;  
  11.   
  12. enum NodeKind{LN, BN};  
  13.   
  14. /** 
  15. *Trie node 
  16. */  
  17. class TrieNode  
  18. {  
  19.     char key;  
  20.     TrieNode[] points = null;  
  21.     NodeKind kind = null;  
  22. }  
  23.   
  24. /** 
  25. * Branch node 
  26. */  
  27. class BranchNode extends TrieNode  
  28. {  
  29.     BranchNode(char k)  
  30.     {  
  31.         super.key = k;  
  32.         super.kind = NodeKind.BN;  
  33.         super.points = new TrieNode[27];  
  34.     }  
  35. }  
  36.   
  37.   
  38. /** 
  39. * Leaf node 
  40. */  
  41. class LeafNode extends TrieNode  
  42. {  
  43.     LeafNode(char k)  
  44.     {  
  45.         super.key = k;  
  46.         super.kind = NodeKind.LN;  
  47.     }  
  48. }  
  49.   
  50.   
  51. public class StandardTrie  
  52. {  
  53.     //Create root node  
  54.     TrieNode root = new BranchNode(‘ ‘);  
  55.   
  56.     //[1].Insert a word into tire tree  
  57.     public void insert(String words)  
  58.     {  
  59.         TrieNode curNode = root;  
  60.         //add ‘$‘ as an end symbol  
  61.         words = words + "$";  
  62.         char[] chars = words.toCharArray();  
  63.           
  64.         for(int i = 0; i < chars.length; i++)  
  65.         {  
  66.             if(chars[i] == ‘$‘)  
  67.             {  
  68.                 curNode.points[26] = new LeafNode(‘$‘);  
  69.             }  
  70.             else  
  71.             {  
  72.                 int pSize = chars[i] - ‘a‘;  
  73.                 // If not exists creat a new branch node  
  74.                 if(curNode.points[pSize] == null)  
  75.                 {  
  76.                     curNode.points[pSize] = new BranchNode(chars[i]);  
  77.                 }  
  78.                 curNode = curNode.points[pSize];  
  79.             }  
  80.         }  
  81.     }  
  82.       
  83.     //[2].Check if a word is in tire tree  
  84.     public boolean fullMatch(String words)  
  85.     {  
  86.         TrieNode curNode = root;  
  87.         char[] chars = words.toCharArray();  
  88.           
  89.         for(int i = 0; i < chars.length; i++)  
  90.         {  
  91.             int pSize = chars[i] - ‘a‘;  
  92.             System.out.print(chars[i]+"->");  
  93.             if(curNode.points[pSize] == null)  
  94.                 return false;  
  95.             curNode = curNode.points[pSize];  
  96.         }  
  97.           
  98.         if(curNode.points[26] != null && curNode.points[26].key == ‘$‘)  
  99.             return true;  
  100.           
  101.         return false;     
  102.     }  
  103.       
  104.       
  105.     //[3].preorder root traverse  
  106.     private void preorderTraverse(TrieNode curNode)  
  107.     {  
  108.         if(curNode != null)  
  109.         {  
  110.             System.out.print(curNode.key);  
  111.               
  112.             if(curNode.kind == NodeKind.BN)  
  113.             {  
  114.                 for(TrieNode node : curNode.points)  
  115.                 {  
  116.                     preorderTraverse(node);  
  117.                 }  
  118.             }  
  119.             else  
  120.                 System.out.println();  
  121.         }  
  122.     }  
  123.       
  124.     //[4].Get root node  
  125.     public TrieNode getRoot()  
  126.     {  
  127.         return this.root;  
  128.     }  
  129.       
  130.       
  131.     public static void main(String[] args)  
  132.     {  
  133.         StandardTrie trie = new StandardTrie();  
  134.           
  135.         trie.insert("amazon");  
  136.         trie.insert("yahoo");  
  137.         trie.insert("haiyun");  
  138.         trie.insert("baidu");  
  139.         trie.insert("alibaba");  
  140.         trie.insert("offer");  
  141.         trie.insert("stock");  
  142.         trie.insert("stop");  
  143.           
  144.         trie.preorderTraverse(trie.getRoot());  
  145.           
  146.         System.out.println(trie.fullMatch("yahoo"));  
  147.         System.out.println(trie.fullMatch("yaho"));  
  148.         System.out.println(trie.fullMatch("baidu"));  
  149.         System.out.println(trie.fullMatch("alibaba"));  
  150.     }  
  151. }  

Trie树(转)