[hihoCoder java] Trie数（字典树）

经典的字典树问题，把AC的代码放在这，作为备忘录。因为对这些经典的问题并不是很熟悉，所以要记录下来。

import java.util.Scanner;

public class Main {


private  final int SIZE=26;
private TrieNode rootNode;//字典树的根
/**
* @param args
*/
Main(){//初始化字典树 
rootNode=new TrieNode();
}

private class TrieNode{//字典树节点
private int num;//有多少单词通过这个节点,即节点字符出现的次数
private TrieNode[] son;//所有的儿子节点
private boolean isEnd;//是不是最后一个节点
private char val;//节点的值

public TrieNode() {
// TODO Auto-generated constructor stub
num=1;
son=new TrieNode[SIZE];
isEnd=false;
}
}
//建立字典树
public void insert(String str){
if(str==null||str.length()==0){
return;
}
TrieNode node=rootNode;
for(int i=0,length=str.length();i<length;i++){
int pos=str.charAt(i)-'a';
if(node.son[pos]==null){
node.son[pos]=new TrieNode();
node.son[pos].val=str.charAt(i);
}else{
node.son[pos].num++;
}
node=node.son[pos];
}
node.isEnd=true;
}

//计算单词前缀的数量
public int countPrefix(String prefix){
if(prefix==null||prefix.length()==0){
return -1;
}
TrieNode node=rootNode;
for(int i=0,length=prefix.length();i<length;i++){
int pos=prefix.charAt(i)-'a';
if(node.son[pos]==null){
return 0;
}else{
node=node.son[pos];
}
}
return node.num;
}
//在字典树中查找一个完全匹配的单词
public boolean has(String str){
if(str==null||str.length()==0){
return false;
}
TrieNode node=rootNode;
for(int i=0,length=str.length();i<length;i++){
int pos=str.charAt(i)-'a';
if(node.son[pos]!=null){
node=node.son[pos];
}else {
return false;
}
}
return node.isEnd;
}


//前序遍历字典树
public void preTraverse(TrieNode node){
if(node!=null){
System.out.println(node.val+"-");
for(TrieNode child:node.son){
preTraverse(child);
}
}

}

public TrieNode getRoot(){
return this.rootNode;
}
public static void main(String[] args) {
// TODO Auto-generated method stub


Main tree= new Main();
Scanner in=new Scanner(System.in);
int m=Integer.parseInt(in.nextLine());
String[] strs=new String[m];
for(int i=0;i<m;i++){
strs[i]=in.nextLine();
}
int n=Integer.parseInt(in.nextLine());
String[] prefix=new String[n];
for(int i=0;i<n;i++){
prefix[i]=in.nextLine();
}
for(String str:strs){
tree.insert(str);
}
for(String pre :prefix){
int num=tree.countPrefix(pre);
System.out.println(num);
}
}


}