Java for LeetCode 208 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
.
解题思路:
参考百度百科:Trie树
已经给出了详细的代码:
JAVA实现如下:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 | class TrieNode { // Initialize your data structure here. int num; // 有多少单词通过这个节点,即节点字符出现的次数 TrieNode[] son; // 所有的儿子节点 boolean isEnd; // 是不是最后一个节点 char val; // 节点的值 TrieNode() { this .num = 1 ; this .son = new TrieNode[ 26 ]; this .isEnd = false ; } } public class Trie { private TrieNode root; public Trie() { root = new TrieNode(); } public void insert(String word) { if (word == null || word.length() == 0 ) return ; TrieNode node = this .root; char [] letters = word.toCharArray(); for ( int i = 0 ; i < word.length(); 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++; } node = node.son[pos]; } node.isEnd = true ; } // Returns if the word is in the trie. public boolean search(String word) { if (word == null || word.length() == 0 ) { return false ; } TrieNode node = root; char [] letters = word.toCharArray(); for ( int i = 0 ; i < word.length(); i++) { int pos = letters[i] - 'a' ; if (node.son[pos] != null ) { node = node.son[pos]; } else { return false ; } } return node.isEnd; } // Returns if there is any word in the trie // that starts with the given prefix. public boolean startsWith(String prefix) { if (prefix == null || prefix.length() == 0 ) { return false ; } TrieNode node = root; char [] letters = prefix.toCharArray(); for ( int i = 0 ; i < prefix.length(); i++) { int pos = letters[i] - 'a' ; if (node.son[pos] != null ) { node = node.son[pos]; } else { return false ; } } return true ; } } // Your Trie object will be instantiated and called as such: // Trie trie = new Trie(); // trie.insert("somestring"); // trie.search("key"); |
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步