211. 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.
转载于此
思路:采用字典树的结构,寻找是否具有某个单词时,对于可以替代任何字母的’.’点,它往下的所有子节点都可以。因此,26个子节点都要考虑。

 class trie{
     public:
       // char val;
        trie* son[26];
        bool isend;
        trie(){
            memset(son,0,sizeof(son));
            isend=0;
           // val=' ';
        }
        ~trie(){
            for(auto it:son)
            delete it;
        }
    };
class WordDictionary {
public:
    // Adds a word into the data structure.
    public:
    WordDictionary(){
        root=new trie();
    }
    ~WordDictionary(){
        delete root;
    }
    void addWord(string word) {
       // if(search(word)) return;
        trie* cur=root;
        for(auto ch:word)
        {
            int pos=ch-'a';
            if(cur->son[pos]==NULL)
            {
                cur->son[pos]=new trie();
                //cur->son[pos]->val=ch;
            }
            cur=cur->son[pos];
        }
        cur->isend=1;
    }

    // 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)
    {
        return helper(word.c_str(),root);
    }
    private:
     trie* root;
    bool helper(const char* word,trie* root) {
       int i,k;
       trie* cur=root;

       for(i=0;word[i];i++)
       {
           char ch=word[i];
           if(cur&&ch!='.')
           cur=cur->son[ch-'a'];
           else if(cur&&ch=='.')
           {
               trie* tmp=cur;
               for(k=0;k<26;k++)
               {
                   cur=tmp->son[k];
                   if(helper(word+i+1,cur))
                   return 1;
               }
           }else break;
       }
       return cur&&cur->isend;
    }

};

// Your WordDictionary object will be instantiated and called as such:
// WordDictionary wordDictionary;
// wordDictionary.addWord("word");
// wordDictionary.search("pattern");
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值