LeetCode—211. 添加与搜索单词 - 数据结构设计

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
}