java trie_java实现的Trie树数据结构

package com.algorithms;

import java.util.HashMap;

import java.util.Map;

public class Trie_Tree{

/**

* 内部节点类

* @author "zhshl"

* @date2014-10-14

*

*/

private class Node{

private int dumpli_num;该字串的反复数目， 该属性统计反复次数的时候实用,取值为0、1、2、3、4、5……

private int prefix_num;///以该字串为前缀的字串数。 应该包含该字串本身。。！

。！

private Node childs[];此处用数组实现，当然也能够map或list实现以节省空间

private boolean isLeaf;///是否为单词节点

public Node(){

dumpli_num=0;

prefix_num=0;

isLeaf=false;

childs=new Node[26];

}

}

private Node root;///树根

public Trie_Tree(){

///初始化trie 树

root=new Node();

}

/**

* 插入字串。用循环取代迭代实现

* @param words

*/

public void insert(String words){

insert(this.root, words);

}

/**

* 插入字串，用循环取代迭代实现

* @param root

* @param words

*/

private void insert(Node root,String words){

words=words.toLowerCase();转化为小写

char[] chrs=words.toCharArray();

for(int i=0,length=chrs.length; i

///用相对于a字母的值作为下标索引，也隐式地记录了该字母的值

int index=chrs[i]-'a';

if(root.childs[index]!=null){

已经存在了，该子节点prefix_num++

root.childs[index].prefix_num++;

}else{

///假设不存在

root.childs[index]=new Node();

root.childs[index].prefix_num++;

}

///假设到了字串结尾，则做标记

if(i==length-1){

root.childs[index].isLeaf=true;

root.childs[index].dumpli_num++;

}

///root指向子节点，继续处理

root=root.childs[index];

}

}

/**

* 遍历Trie树，查找全部的words以及出现次数

* @return HashMap map

*/

public HashMap getAllWords(){

//HashMap map=new HashMap();

return preTraversal(this.root, "");

}

/**

* 前序遍历。。。

* @param root子树根节点

* @param prefixs查询到该节点前所遍历过的前缀

* @return

*/

private HashMap preTraversal(Node root,String prefixs){

HashMap map=new HashMap();

if(root!=null){

if(root.isLeaf==true){

当前即为一个单词

map.put(prefixs, root.dumpli_num);

}

for(int i=0,length=root.childs.length; i

if(root.childs[i]!=null){

char ch=(char) (i+'a');

递归调用前序遍历

String tempStr=prefixs+ch;

map.putAll(preTraversal(root.childs[i], tempStr));

}

}

}

return map;

}

/**

* 推断某字串是否在字典树中

* @param word

* @return true if exists ,otherwise false

*/

public boolean isExist(String word){

return search(this.root, word);

}

/**

* 查询某字串是否在字典树中

* @param word

* @return true if exists ,otherwise false

*/

private boolean search(Node root,String word){

char[] chs=word.toLowerCase().toCharArray();

for(int i=0,length=chs.length; i

int index=chs[i]-'a';

if(root.childs[index]==null){

///假设不存在，则查找失败

return false;

}

root=root.childs[index];

}

return true;

}

/**

* 得到以某字串为前缀的字串集。包含字串本身。 相似单词输入法的联想功能

* @param prefix 字串前缀

* @return 字串集以及出现次数，假设不存在则返回null

*/

public HashMap getWordsForPrefix(String prefix){

return getWordsForPrefix(this.root, prefix);

}

/**

* 得到以某字串为前缀的字串集。包含字串本身。

* @param root

* @param prefix

* @return 字串集以及出现次数

*/

private HashMap getWordsForPrefix(Node root,String prefix){

HashMap map=new HashMap();

char[] chrs=prefix.toLowerCase().toCharArray();

for(int i=0, length=chrs.length; i

int index=chrs[i]-'a';

if(root.childs[index]==null){

return null;

}

root=root.childs[index];

}

///结果包含该前缀本身

///此处利用之前的前序搜索方法进行搜索

return preTraversal(root, prefix);

}

}