211. 添加与搜索单词 - 数据结构设计
题目描述:请你设计一个数据结构,支持 添加新单词 和 查找字符串是否与任何先前添加的字符串匹配 。
实现词典类 WordDictionary :
WordDictionary() 初始化词典对象
void addWord(word) 将 word 添加到数据结构中,之后可以对它进行匹配
bool search(word) 如果数据结构中存在字符串与 word 匹配,则返回 true ;否则,返回 false 。word 中可能包含一些 ‘.’ ,每个 . 都可以表示任何一个字母。
考察重点:同207,在基础上增加通配符’.’ (只需要对通配符节点遍历其26个子节点即可)
type WordDictionary struct {
Next []*WordDictionary
Endflag bool
}
func Constructor3() WordDictionary {
return WordDictionary{Next: make([]*WordDictionary, 26), Endflag: false}
}
func (wd *WordDictionary) AddWord(word string) {
for i := 0; i < len(word); i++ {
if wd.Next[word[i]-'a'] == nil {
temp := Constructor3()
wd.Next[word[i]-'a'] = &temp
}
wd = wd.Next[word[i]-'a']
}
wd.Endflag = true
}
func (wd *WordDictionary) Search(word string) bool {
for i := 0; i < len(word); i++ {
// fmt.Println(word[i])
if word[i] == 46 { //这里比207要多一步,用来判断作为通配符的'.'的特殊情况
for j := 0; j < 26; j++ { //遍历26个子节点,只要有一个可以完成search,则返回true
if wd.Next[j] != nil && wd.Next[j].Search(word[i+1:]) { //用子节点和剩余字符串完成余下的search
return true
}
}
return false
}
if wd.Next[word[i]-'a'] == nil {
return false
}
wd = wd.Next[word[i]-'a']
}
return wd.Endflag
}