|
| 1 | +/* |
| 2 | + * @lc app=leetcode id=1032 lang=java |
| 3 | + * |
| 4 | + * [1032] Stream of Characters |
| 5 | + */ |
| 6 | + |
| 7 | +// @lc code=start |
| 8 | + |
| 9 | +/* |
| 10 | +将dict中的单词逆序存入一个trie中,query的时候倒序在trie中搜索 |
| 11 | +time: O(n * m) n: number of words m: length of word |
| 12 | +space: O(n * m) |
| 13 | +*/ |
| 14 | +class StreamChecker { |
| 15 | + class TrieNode { |
| 16 | + TrieNode[] children; |
| 17 | + boolean isWord; |
| 18 | + public TrieNode() { |
| 19 | + children = new TrieNode[26]; |
| 20 | + isWord = false; |
| 21 | + } |
| 22 | + } |
| 23 | + |
| 24 | + TrieNode root; |
| 25 | + StringBuilder sb; |
| 26 | + public StreamChecker(String[] words) { |
| 27 | + root = new TrieNode(); |
| 28 | + buildTrie(words); |
| 29 | + sb = new StringBuilder(); |
| 30 | + } |
| 31 | + |
| 32 | + public boolean query(char letter) { |
| 33 | + sb.append(letter); |
| 34 | + TrieNode cur = root; |
| 35 | + for (int i = sb.length() - 1; i >= 0 && cur != null; --i) { |
| 36 | + cur = cur.children[sb.charAt(i) - 'a']; |
| 37 | + if (cur != null && cur.isWord) { |
| 38 | + return true; |
| 39 | + } |
| 40 | + } |
| 41 | + return false; |
| 42 | + } |
| 43 | + |
| 44 | + private void buildTrie(String[] words) { |
| 45 | + for (String word : words) { |
| 46 | + TrieNode cur = root; |
| 47 | + // build Trie in reverse order |
| 48 | + for (int i = word.length() - 1; i >= 0; --i) { |
| 49 | + if (cur.children[word.charAt(i) - 'a'] == null) { |
| 50 | + cur.children[word.charAt(i) - 'a'] = new TrieNode(); |
| 51 | + } |
| 52 | + cur = cur.children[word.charAt(i) - 'a']; |
| 53 | + } |
| 54 | + cur.isWord = true; |
| 55 | + } |
| 56 | + } |
| 57 | +} |
| 58 | + |
| 59 | +/** |
| 60 | + * Your StreamChecker object will be instantiated and called as such: |
| 61 | + * StreamChecker obj = new StreamChecker(words); |
| 62 | + * boolean param_1 = obj.query(letter); |
| 63 | + */ |
| 64 | +// @lc code=end |
| 65 | + |
0 commit comments