Trie(字典树)的递归版本

递归请一定要明确函数的语义

private void insert(Node node,String word,int index)在以node为根的trie树中插入以index开始的子串(word[index,...,end])
private boolean contains(Node node,String word,int index):在以node为根的trie中查询是否存在从index开始的子串(word[index,...,end])
private boolean isPrefix(Node node,String word ,int index):在以node为根的trie中检索是否存在以index开始的子串为前缀(word[index,...,end])

递归版本可以实现更为复杂的逻辑

例如:https://leetcode-cn.com/problems/add-and-search-word-data-structure-design/

class Trie {
        private class Node {
            public boolean isWord;
            public TreeMap<Character, Node> children;

            public Node(boolean isWord) {
                this.isWord = isWord;
                children = new TreeMap<>();
            }

            public Node() {
                this(false);
            }
        }
        private Node root;
        private int size;
        Trie(){
            root=new Node();
            size=0;
        }
        public int getSize(){
            return  size;
        }
        public void insert(String word){
            insert(root,word,0);
        }
        //向以node为根的trie树中添加从index开始的子串(word[index...end])
        private void insert(Node node,String word,int index){
            if(index==word.length()){
                if(!node.isWord){
                    node.isWord=true;
                    size++;
                }
                return;
            }
             char c=word.charAt(index);
            if(node.children.get(c)==null) node.children.put(c,new Node());
            insert(node.children.get(c),word,index+1);
        }
        public boolean contains(String word){
            return contains(root,word,0);
        }
        //在以node为根的trie树汇总查询是否存在以index开始的子串(word[index...end])
        public boolean contains(Node node,String word,int index){
            if(index==word.length()) return node.isWord;
            char c=word.charAt(index);
            if(node.children.get(c)==null) return false;
            return contains(node.children.get(c),word,index+1);
        }
        public  boolean isPrefix(String word){
            return isPrefix(root,word,0);
        }
        //在以node为根的trie树中查询是否以index开始的子串(word[index...end])
        public boolean isPrefix(Node node,String word,int index){
            if(index==word.length()) return true;
            char c=word.charAt(index);
            if(node.children.get(c)==null) return  false;
            return isPrefix(node.children.get(c),word,index+1);
        }
    }
  • 3
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值