LeetCode: Add and Search Word - Data structure design

Design a data structure that supports the following two operations:

void addWord(word)
bool search(word)

search(word) can search a literal word or a regular expression string containing only letters a-z or .. A . means it can represent any one letter.

For example:

addWord("bad")
addWord("dad")
addWord("mad")
search("pad") -> false
search("bad") -> true
search(".ad") -> true
search("b..") -> true

Note:

You may assume that all words are consist of lowercase letters a-z.

class TrieNode {
public:
    // Initialize your data structure here.
    TrieNode(): child({NULL}), isEnd(false)
    {
        
    }
    bool isEnd;
    TrieNode *child[26];
};
class WordDictionary {
public:

    // Adds a word into the data structure.
	WordDictionary()
	{
		root = new TrieNode();

	}

    void addWord(string word) {
        TrieNode *iter = root;
        for(int i = 0; i < word.size(); i++)
        {
            if(iter->child[word[i] - 'a'] != NULL)
            {
                iter = iter->child[word[i] - 'a'];
            }
            else
            {
                TrieNode *temp = new TrieNode();
                iter->child[word[i] - 'a'] = temp;
                iter = temp;
            }
        }
        iter->isEnd = true;
    }

    // Returns if the word is in the data structure. A word could
    // contain the dot character '.' to represent any one letter.
    bool search(string word) {
		TrieNode *iter = root;
		return dfs(word, iter);
    }
private:
	bool dfs(string &word, TrieNode *iter)
	{
		if(word.empty())
			return iter->isEnd;
		if('.' == word[0])
		{
			bool result = false;
			for(int j = 0; j < 26; j++)
			{
				if(NULL == iter->child[j])
					continue;
				string sub = word.size() <= 1 ? "" : word.substr(1);
				result |= dfs(sub, iter->child[j]);
				if(true == result)
					return true;
			}
			return false;
		}
        else if(iter->child[word[0] - 'a'] != NULL)
        {
            string sub = word.size() <= 1 ? "" : word.substr(1);
			return dfs(sub, iter->child[word[0] - 'a']);
        }
        else
        {
            return false;
        }

	}
    TrieNode* root;
};



  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器学习模型机器
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值