前缀树Java

最新推荐文章于 2024-07-10 12:15:00 发布

在学了在学了！

最新推荐文章于 2024-07-10 12:15:00 发布

阅读量167

点赞数

分类专栏：笔记文章标签： java 算法 c++

本文链接：https://blog.csdn.net/xieyihu/article/details/126086965

版权

笔记专栏收录该内容

20 篇文章 0 订阅

订阅专栏

一、前缀树

 public static class TireNode{
        public int pass;//经过这个路径的次数
        public int end;//字符串的结尾
        
        public TireNode[] nexts;//以数组的形式或者采用以下两种
        /*
        HashMap<char,Node> nexts;
        TreeMap<char,Node> nexts;
         */
        public TireNode(){
            pass = 0;
            end = 0;
            nexts = new TireNode[26];//用0-26代替26个字母
        }
    }
    
    public static class Tire{
        private TireNode root;
        
        public Tire(){
            root = new TireNode();
        }
        
        public void insert(String word){
            if(word == null){
                return;
            }
            char[] chs = word.toCharArray();
            TireNode node = root;
            node.pass++;
            int index = 0;
            for(int i = 0; i < chs.length; i++){
                index = chs[i] - 'a';//由字符对应走哪条路
                if (node.nexts[index] == null){
                    node.nexts[index] = new TireNode();
                }
                node = node.nexts[index];
                node.pass++;
            }
            node.end++;
        }
    }

二、word之前加入过几次

public int search(String word){
        if (word == null){
            return  0;
        }
        char[] chs = word.toCharArray();
        TireNode node = root;
        int index = 0;
        for(int i = 0; i < chs.length; i++){
            index = chs[i] - 'a';
            if (node.nexts[index] == null){
                return 0;
            }
        }
        return node.end;
    }

三、所有加入过的字符串中，使用pre作为前缀的有几个

public int preNumber(String pre){
        if(pre == null){
            return 0;
        }
        char[] chs = pre.toCharArray();
        TireNode node = root;
        int index = 0;
        for(int i = 0; i < chs.length; i++){
            index = chs[i] - 'a';
            if(node.nexts[index] == null){
                return 0;
            }
            node = node.nexts[index];
        }
        return node.pass;
    }

四、删除

public void delete(String word){
        if (search(word) != 0){//树中有这个元素
            char[] chs = word.toCharArray();
            TireNode node = root;
            node.pass--;
            int index = 0;
            for (int i = 0; i < chs.length; i++){
                index = chs[i] - 'a';
                if (--node.nexts[index].pass == 0){
                    node.nexts[index] = null;
                }
                node = node.nexts[index];
            }
            node.end--;
        }
    }