Trie tree(字典树)的Java实现及其应用-统计以某字符串为前缀的单词的数量

import java.util.LinkedList;

public class CaseInsensitiveTrie {

	/**
	字典树的Java实现。实现了插入、查询以及深度优先遍历。 
    Trie tree's java implementation.(Insert,Search,DFS)
    
	Problem Description
	Ignatius最近遇到一个难题,老师交给他很多单词(只有小写字母组成,不会有重复的单词出现),现在老师要他统计出以某个字符串为前缀的单词数量(单词本身也是自己的前缀).

	Input
	输入数据的第一部分是一张单词表,每行一个单词,单词的长度不超过10,它们代表的是老师交给Ignatius统计的单词,一个空行代表单词表的结束.第二部分是一连串的提问,每行一个提问,每个提问都是一个字符串.
	注意:本题只有一组测试数据,处理到文件结束.

	Output
	对于每个提问,给出以该字符串为前缀的单词的数量.

	Sample Input
	banana
	band
	bee
	absolute
	acm

	ba
	b
	band
	abc
	 
	Sample Output
	2
	3
	1
	0
	*/
	private int SIZE = 26;// 26 letters.CaseInsensitive.
	private TrieNode root;

	CaseInsensitiveTrie() {
		root = new TrieNode();
	}

	private class TrieNode {
		private int num;// how many words go through this node.
		private TrieNode[] son;// TrieNode[26] in this case
		private boolean isEnd;// end of a string.
		private char val;// No this field mostly.
		// But I think it's easy to traverse when having a specific value in
		// each node.
		private boolean visited;// add this field for DFS

		TrieNode() {
			num = 1;//num=0 is wrong
			son = new TrieNode[SIZE];
			isEnd = false;
			visited = false;
		}
	}

	public void insert(String str) {
		if (str == null || str.length() == 0) {
			return;
		}
		TrieNode node = root;
		char[] letters=str.toCharArray();
		for (int i = 0, len = str.length(); i < len; i++) {
			int pos = letters[i] - 'a';
			if (node.son[pos] == null) {
				node.son[pos] = new TrieNode();
				node.son[pos].val = letters[i];
			} else {
				node.son[pos].num++;//for 'countPrefix(String prefix)'
			}
			node = node.son[pos];
		}
		node.isEnd = true;
	}

	/**
	 * count how many words start with the specific prefix.
	 * since we count it in insertion,what we need to do is to return the 'num' of the last letter of prefix.
	 */
	public int countPrefix(String prefix){
		if(prefix==null||prefix.length()==0){
			return -1;
		}
		TrieNode node=root;
		char[] letters=prefix.toCharArray();
		for(int i=0,len=prefix.length();i<len;i++){
			int pos=letters[i]-'a';
			if(node.son[pos]==null){
				return 0;
			}else{
				node=node.son[pos];
			}
		}
		return node.num;
	}
	
	// search a word in the tree.Complete matching.
	public boolean has(String str) {
		if (str == null || str.length() == 0) {
			return false;
		}
		TrieNode node = root;
		char[] letters=str.toCharArray();
		for (int i = 0, len = str.length(); i < len; i++) {
			int pos = letters[i] - 'a';
			if (node.son[pos] != null) {
				node = node.son[pos];
			} else {
				return false;
			}
		}
		return node.isEnd;
	}

	// DFS.Use stack.Like a 26-nary tree.
	public void printAllWords() {
		TrieNode rootNode = root;
		if (root == null){
			return;
		}
		LinkedList<TrieNode> list = new LinkedList<TrieNode>();
		for (int i = 0; i < SIZE; i++) {
			TrieNode node = rootNode.son[i];
			if (node != null) {
				list.addLast(node);
				while (!list.isEmpty()) {
					TrieNode curNode = list.getLast();
					TrieNode firstChild = firstChildOf(curNode);
					while (firstChild != null) {
						list.addLast(firstChild);
						firstChild = firstChildOf(firstChild);// DFS.
					}
					TrieNode strEnd = list.getLast();
					if (strEnd.isEnd) {
						printLinkedList(list);
					}
					list.removeLast();
				}
			}
			list.clear();
		}
	}

	//print each node in preOrder.
	public void preTraverse(TrieNode node){
		if(node!=null){
			System.out.print(node.val+"-");
			for(TrieNode child:node.son){
				preTraverse(child);
			}
		}
		
	}
	// get the first unvisited child of a node.
	public TrieNode firstChildOf(TrieNode node) {
		if (node == null)
			return null;
		for (int i = 0; i < SIZE; i++) {
			TrieNode tmp = node.son[i];
			if (tmp != null && !tmp.visited) {
				tmp.visited = true;
				return tmp;
			}
		}
		return null;
	}

	public static void printLinkedList(LinkedList<TrieNode> list) {
		if (list == null || list.size() == 0){
			return;
		}
		StringBuilder sb = new StringBuilder();
		for (int i = 0, len = list.size(); i < len; i++) {
			sb.append(list.get(i).val);
		}
		System.out.println(sb.toString());
	}

	public TrieNode getRoot(){
		return this.root;
	}
	
	public static void main(String[] args) {
		CaseInsensitiveTrie tree = new CaseInsensitiveTrie();
		String[] strs={
				"banana",
				"band",
				"bee",
				"absolute",
				"acm",
		};
		String[] prefix={
			"ba",
			"b",
			"band",
			"abc",
		};
		for(String str:strs){
			tree.insert(str);
		}
		System.out.println(tree.has("abc"));
		tree.preTraverse(tree.getRoot());
		System.out.println();
		tree.printAllWords();
		for(String pre:prefix){
			int num=tree.countPrefix(pre);
			System.out.println(pre+" "+num);
		}
		
	}
}

另一道类似的题目：

Description
Given a list of phone numbers, determine if it is consistent in the sense that no number is the prefix of another. Let's say the phone catalogue listed these numbers:
Emergency 911
Alice 97 625 999
Bob 91 12 54 26
In this case, it's not possible to call Bob, because the central would direct your call to the emergency line as soon as you had dialled the first three digits of Bob's phone number. So this list would not be consistent.
Input
The first line of input gives a single integer, 1 ≤ t ≤ 40, the number of test cases. Each test case starts with n, the number of phone numbers, on a separate line, 1 ≤ n ≤ 10000. Then follows n lines with one unique phone number on each line. A phone number is a sequence of at most ten digits.
Output
For each test case, output "YES" if the list is consistent, or "NO" otherwise.
Sample Input
2
3
911
97625999
91125426
5
113
12340
123440
12345
98346
Sample Output
NO
YES

这道题用TrieTree也好做：在插入某个号码时候，如果遇到一个节点已经isEnd==true而号码还没有插入完成，则返回NO

评论

1 楼 neyshule 2012-09-04  

不错啊，有没有删除的代码啊。