leetcode--Implement Trie (Prefix Tree)

Implement a trie with insert, search, and startsWith methods.

Note:
You may assume that all inputs are consist of lowercase letters a-z.

[java]  view plain  copy

1. class TrieNode {  
2.     // Initialize your data structure here.  
3.     public TrieNode() {}  
4.     Map<Character,TrieNode> next = new HashMap<Character,TrieNode>();  
5.     char c='\0';  
6.     boolean isEnd = false;  
7.     public TrieNode(char c) {  
8.         this.c = c;          
9.     }  
10. }  
11. public class Trie {  
12.     private TrieNode root;  
13.   
14.     public Trie() {  
15.         root = new TrieNode();  
16.     }  
17.   
18.     // Inserts a word into the trie.  
19.     public void insert(String word) {  
20.         TrieNode cur = root;  
21.         for(int i=0;i<word.length();i++){  
22.             if(!cur.next.containsKey(word.charAt(i))){               
23.                 cur.next.put(word.charAt(i), new TrieNode(word.charAt(i)));               
24.             }  
25.             cur = cur.next.get(word.charAt(i));  
26.         }   
27.         cur.isEnd = true;  
28.     }  
29.   
30.     // Returns if the word is in the trie.  
31.     public boolean search(String word) {  
32.         TrieNode cur = root;  
33.         for(int i=0;i<word.length();i++){  
34.             if(cur.next.containsKey(word.charAt(i))){  
35.                 cur = cur.next.get(word.charAt(i));  
36.             }else{  
37.                 return false;  
38.             }  
39.         }  
40.         if(cur.isEnd) return true;  
41.         return false;  
42.     }  
43.   
44.     // Returns if there is any word in the trie  
45.     // that starts with the given prefix.  
46.     public boolean startsWith(String prefix) {  
47.         TrieNode cur = root;  
48.         for(int i=0;i<prefix.length();i++){  
49.             if(cur.next.containsKey(prefix.charAt(i))){  
50.                 cur = cur.next.get(prefix.charAt(i));  
51.             }else{  
52.                 return false;  
53.             }  
54.         }  
55.         return true;      
56.     }  
57. }  
58.   
59. // Your Trie object will be instantiated and called as such:  
60. // Trie trie = new Trie();  
61. // trie.insert("somestring");  
62. // trie.search("key");  

原文链接http://blog.youkuaiyun.com/crazy__chen/article/details/46575533