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.

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);
        }
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值