码迷,mamicode.com
首页 > 其他好文 > 详细

Add and Search Word - Data structure design

时间:2015-05-16 20:38:35      阅读:128      评论:0      收藏:0      [点我收藏+]

标签:leetcode   211   add   search   word   

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.

click to show hint.

You should be familiar with how a Trie works. If not, please work on this problem: Implement Trie (Prefix Tree) first.

思路:
该题目和”Trie”相似,稍微有些变化。

public class WordDictionary {
    static class Node{
        public Map<Character,Node> map;
        public char c;
        public boolean isTail;
        public Node(char c){
            this.c=c;
            map = new HashMap<>();
            isTail=false;
        }

    }

    private Node root;

    public WordDictionary(){
        root = new Node((char)0);
    }

    // Adds a word into the data structure.
    public void addWord(String word) {
        addWord(word,root);
    }

    private void addWord(String word,Node root){
        if(word.length()==0){
            root.isTail=true;
            return;
        }
        char c=word.charAt(0);
        if(!root.map.containsKey(c)){
            Node n = new Node(c);
            root.map.put(c,n);
        }
        Node n=root.map.get(c);
        addWord(word.substring(1),n);
    }

    // Returns if the word is in the data structure. A word could
    // contain the dot character ‘.‘ to represent any one letter.
    public boolean search(String word) {
        return search(word,root);
    }

    private boolean search(String word,Node root){
        if(word.length()==0){
            return root.isTail==true; 
        }
        char c=word.charAt(0);
        if(c==‘.‘){//深度优先搜索
            for(Map.Entry<Character,Node> en:root.map.entrySet()){
                boolean ret = search(word.substring(1),en.getValue());
                if(ret){
                    return true;
                }
            }
            return false;
        }else{
            if(!root.map.containsKey(c)){
                return false;
            }
            Node n=root.map.get(c);
            return search(word.substring(1),n);
        }
    }
}

Add and Search Word - Data structure design

标签:leetcode   211   add   search   word   

原文地址:http://blog.csdn.net/u010786672/article/details/45770709

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!