添加与搜索单词 - 数据结构设计
请你设计一个数据结构,支持 添加新单词 和 查找字符串是否与任何先前添加的字符串匹配 。
实现词典类 WordDictionary :
- WordDictionary() 初始化词典对象
- void addWord(word) 将 word 添加到数据结构中,之后可以对它进行匹配
- bool search(word) 如果数据结构中存在字符串与 word 匹配,则返回 true ;否则,返回 false 。word 中可能包含一些 ‘.’ ,每个 . 都可以表示任何一个字母。
示例:
输入:
[“WordDictionary”,“addWord”,“addWord”,“addWord”,“search”,“search”,“search”,“search”]
[[],[“bad”],[“dad”],[“mad”],[“pad”],[“bad”],[“.ad”],[“b…”]]
输出:
[null,null,null,null,false,true,true,true]
解释:
WordDictionary wordDictionary = new WordDictionary();
wordDictionary.addWord(“bad”);
wordDictionary.addWord(“dad”);
wordDictionary.addWord(“mad”);
wordDictionary.search(“pad”); // 返回 False
wordDictionary.search(“bad”); // 返回 True
wordDictionary.search(“.ad”); // 返回 True
wordDictionary.search(“b…”); // 返回 True
解题思路
字典树的构建:
- 每个节点包含一个数组或哈希表,表示从当前节点可以到达的下一个字符。
- 用一个布尔变量表示是否到达一个单词的结尾。
添加单词:
- 从根节点开始,逐字符插入,如果字符不存在于当前节点的子节点中,则创建一个新的子节点。
搜索单词:
- 对于常规字符,按字符匹配进行搜索。
- 对于通配符 .,需要递归地尝试匹配所有可能的子节点。
Java实现
public class WordDictionary {
private TrieNode root;
// 字典树节点类
private class TrieNode {
TrieNode[] children;
boolean isEnd;
public TrieNode() {
children = new TrieNode[26];
isEnd = false;
}
}
public WordDictionary() {
root = new TrieNode();
}
// 添加单词
public void addWord(String word) {
TrieNode node = root;
for (char c : word.toCharArray()) {
if (node.children[c - 'a'] == null) {
node.children[c - 'a'] = new TrieNode();
}
node = node.children[c - 'a'];
}
node.isEnd = true;
}
// 搜索字符, '.' 代表任意字符
public boolean search(String word) {
return searchInNode(word, root);
}
//搜索字符实现类
private boolean searchInNode(String word, TrieNode node) {
for (int i = 0; i < word.length(); i++) {
char c = word.charAt(i);
if (c == '.') {
for (TrieNode child : node.children) {
if (child != null && searchInNode(word.substring(i + 1), child)) {
return true;
}
}
return false;
} else {
if (node.children[c - 'a'] == null) {
return false;
}
node = node.children[c - 'a'];
}
}
return node.isEnd;
}
public static void main(String[] args) {
WordDictionary wordDictionary = new WordDictionary();
wordDictionary.addWord("bad");
wordDictionary.addWord("dad");
wordDictionary.addWord("mad");
System.out.println(wordDictionary.search("pad")); // false
System.out.println(wordDictionary.search("bad")); // true
System.out.println(wordDictionary.search(".ad")); // true
System.out.println(wordDictionary.search("b..")); // true
}
}
时间空间复杂度
-
时间复杂度:
addWord:O(L),其中 L 是单词的长度。
search:最坏情况下 O(N * 26^L),其中 N 是节点数量,L 是单词的长度 -
空间复杂度:O(N * L),其中 N 是单词数量,L 是平均单词长度。