20-Trie

  • 需求

如何判断一堆不重复的字符串,是否已某个前缀开头?

我们可以用Set/Map来存储字符串,然后遍历所有的字符串进行判断,通过这种方式实现,时间复杂度为O(n)

我们有没有更优的数据结构来实现前缀搜索?

Trie就可以。

  • Trie

Trie也叫做字典树,前缀树(Prefix Tree),单词查找树

Trie搜索字符串的效率主要跟字符串的长度有关。

假设使用Trie存储cat,dog,doggy,does,cast,add六个单词,那它是怎么存储的呢?

首先,存储cat是将单词拆分为单个字符进行存储,如

如果要存储dog,怎么办呢?

然后存储doggy,在存储这个单词时,发现dog是当前单词的前缀,那只需要在原来dog的基础上,再加一个g,y即可

然后再存储does,同样的,利用已有的前缀进行存储,在已有的基础上,增加自己不同的字符

同理,存储cast

存储add

最终,所有的节点存储完成

当我们要搜索某个前缀时,我们只需要从根节点开始,一个字符一个字符的匹配即可。这样可以大大的提高搜索的效率。

  • Trie的实现

接口设计

第一种接口设计

int size();
boolean isEmpty();
void add(String str);
void remove(String str);
boolean startsWith(String prefix);
boolean contains(String str);
void clear();

第二种接口设计

int size();
boolean isEmpty();
V add(String str,V value);
V remove(String str);
V get(String str)
boolean startsWith(String prefix);
boolean contains(String str);
void clear();

第二种接口,在存储一个一个字符串的同时,可以存储一个值,由于第二种接口设计的实现包含了第一种接口设计,那么我们在这里就使用第二种方式来进行实现。具体实现可以查看demo中的源码

删除的注意点

我们在删除节点时,需要注意,

  1. 我们往上删除是,如果有某个单词是当前被删除单词的前缀时,就不要再往前删了,例如上面的单词中,我们删除dog单词,我们不能讲这三个节点从树中删除掉,因此我们需要判断我当前要删除的单词,最后一个节点,是否还有子节点,如果有,就只需要将最后的字符标记不为单词结尾就可以了。
    删除前:


    删除后:



    这样我们就把dog删除掉了

  2. 如果没有子节点,例如d,t,t,y,s这些节点,这些节点如果没有子节点,就直接从后面往前删除。但是在往上删除的时候需要注意,如果当前节点删除完子节点以后,发现还有其他子节点,就不要再往上删除了;例如删除does的时候,当我们删除掉字符e时,就不能再往上删除了。即从最后开始往回删除,直到某节点删除子节点后,还有子节点时,就停止删除,单词的删除就结束了。

删除对应代码实现为:

V remove(String str) {
    //找到最后一个节点
    Node<V> node = node(str);
    V oldValue = node.value;
    //如果不是单词结尾,不用做任何处理
    if (node == null || !node.word) return null;
    size--;
    //如果还有子节点
    if (node.children != null && !node.children.isEmpty()) {
        node.word = false;//把当前节点的单词结尾标记去掉
        node.value = null;
        return oldValue;
    }
    //如果没有子节点
    Node<V> parent = null;
    while ((parent = node.parent) != null) {
        parent.children.remove(node.character);
        if (parent.word || !parent.children.isEmpty()) {//删掉以后还有其他子节点,就不再往上删除,停止循环
            break;
        }
        node = parent;
    }
    return oldValue;
}
  • 总结

Trie的优点:搜索前缀的效率主要跟前缀的长度有关

Trie的缺点:需要消耗大量的内存,因此还有待改进

更多Trie相关的数据结构和算法

  • Double-array Trie
  • Suffix Tree
  • Patricia Tree
  • Crit-bit Tree
  • AC自动机

GitHub地址

本节完!

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 声明:摘自github:https://github.com/ZtesoftCS/go-ethereum-code...
    蓝Renly阅读 811评论 0 0
  • 什么是“Trie树”? Trie树,又称前缀树或字典树,是一种有序树,用于保存关联数组,其中的键通常是字符串。与二...
    尼桑麻阅读 870评论 0 2
  • 引言 go-etherum的包trie实现了Merkle Patricia Tries,这里用简称MPT来称呼这种...
    泡泡龙吐泡泡阅读 3,472评论 0 1
  • 我是一名C++软件开发程序员,今年的5月份,在一个长沙的区块链活动上认识、了解了初链,这个区块链公链项目。在活动中...
    小绍阅读 685评论 0 0
  • 今日我虽然不在家里。但家里一切因着科技发达,我什么都看得到。今天我的家里特别热闹,堂哥一家,表姐一家,还有...
    等等的小花妹妹幸福呀阅读 801评论 0 0