数据结构必知 --- 前缀树

简介: 数据结构必知 --- 前缀树

写在前


  • 什么是字典树?
  1. Trie树,即字典树,又称单词查找树或键树,是一种树形结构,是一种哈希树的变种。Trie 一词来自 retrieval,发音为 /tri:/ "tree",也有人读为 /traɪ/ "try"。
  2. 典型应用是用于统计和排序大量的字符串(但不仅限于字符串),所以经常被搜索引擎系统用于文本词频统计。它的优点是:最大限度地减少无谓的字符串比较。
  • Trie的核心思想是空间换时间,利用字符串的公共前缀来降低查询时间的开销以达到提高效率的目的。
  • 基本性质
  1. 根节点不包含字符,除根节点外每一个节点都只包含一个字符。
  2. 从根节点到某一节点,路径上经过的字符连接起来,为该节点对应的字符串。
  3. 每个节点的所有子节点包含的字符都不相同。
  • 复杂度分析
    1.插入和查询的效率很高,时间复杂度都为O(m),其中 m 是待插入/查询的字符串的长度
    注:关于查询,会有人说 hash 表时间复杂度是O(1)不是更快?但是,哈希搜索的效率通常取决于 hash 函数的好坏,若一个坏的 hash 函数导致很多的冲突,效率并不一定比Trie树高。
    2.最坏的空间复杂度实现方式可以到S^L,其中S为字符集L为最长长度
  • Trie树的应用
  • 字符串检索:事先将已知的一些字符串(字典)的有关信息保存到 Trie 里,查找另外一些未知字符串是否出现过或者出现频率。
  • 字符串最长公共前缀:Trie 利用多个字符串的公共前缀来节省存储空间,反之,当我们把大量字符串存储到一棵 Trie 上时,我们可以快速得到某些字符串的公共前缀。
  • 排序:Trie 树是一棵多叉树,只要先序遍历整棵树,输出相应的字符串,便是按字典序排序的结果。
  • 作为其他数据结构和算法的辅助结构:如后缀树,AC自动机等。

基本实现


前缀树创建

假设有b,abc,abd,bcd,abcd,efg,hii 这6个单词,那我们创建trie树就得到。

image.png

可以看出,Trie树的关键字一般都是字符串,而且Trie树把每个关键字保存在一条路径上,而不是一个结点中。另外,两个有公共前缀的关键字,在Trie树中前缀部分的路径相同。

前缀树的主要操作就是插入和查询,即将一个字符串加入集合中和查询集合中有没有这个字符串。每个字符加两个变量,path 定义节点经过的次数,end 定义最后一个插入的节点

public class TrieNode {
    private int path;
    private int end;
    private TrieNode[] nexts;
    public TrieNode() {
        path = 0;
        end = 0;
        nexts = new TrieNode[26];
    }
}

前缀树插入

private TrieNode root;  // 根节点
public TrieNode {
    root = new TrieNode();
}

/**
    往前缀树中插入字符串(这里只包含26个小写字母)
*/
private void insert(String word) {
    if (word == null) {
        return;
    }
    int n = word.length();
    char[] chs = word.toCharArray();
    TrieNode node = root;
    int index = 0;
    for (int i = 0; i < n; i++) {
        index = chs[i] - 'a';
        if (node.nexts[index] == null) {
            node.nexts[index] = new TrieNode();    // 创建当前字符
        }
        node = node.nexts[index];
        node.path++;
    }
    node.end++;
}

前缀树查询

/**
    1、查询前缀树某字符串出现的次数,返回结尾字符出现的次数,即node.end
    2、查询以某字符串为前缀的数量,返回每个字符被划过的次数,即node.path
    如果不存在返回-1,类似插入操作!
*/
private int searchCount(String str) {
    if (str == null) {
        return -1;
    }
    int n = str.length();
    char[] chs = str.toCharArray();
    TrieNode node = root;
    int index = 0;
    for (int i = 0; i < n; i++) {
        index = chs[i] - 'a';
        if (node.nexts[index] == null) {
            return -1;    
        }
        node = node.nexts[index];  
    }
    return node.end;    // 返回以当前关键字结尾的字符串的个数
    //return node.path;  //插字符串时到达几次
}
/**
    3、查询集合中是否存在某个字符串,同理
*/
private boolean search(String str) {
    if (str == null) {
        return false;
    }
    int n = str.length();
    char[] chs = str.toCharArray();
    TrieNode node = root;
    int index = 0;
    for (int i = 0; i < n; i++) {
        index = chs[i] - 'a';
        if (node.nexts[index] == null) {
            return false;    
        }
        node = node.nexts[index];  
    }
    return true;   
}

案例分析


T208 前缀树

代码实现:

class Trie {
    class TrieNode {
        boolean isEnd;
        TrieNode[] nexts = new TrieNode[26];
    }
    TrieNode root;
    /** Initialize your data structure here. */
    public Trie() {
        root = new TrieNode(); 
    }
    /** Inserts a word into the trie. */
    public void insert(String word) {
        if (word == null) {
            return;
        }
        TrieNode node = root;
        int index = 0;
        for (int i = 0; i < word.length(); i++) {
            index = word.charAt(i) - 'a';
            if (node.nexts[index] == null) {
                node.nexts[index] = new TrieNode();
            }
            node = node.nexts[index];
        }
        node.isEnd = true;
    }
    /** Returns if the word is in the trie. */
    public boolean search(String word) {
        if (word == null) {
            return false;
        }
        TrieNode node = root;
        int index = 0;
        for (int i = 0; i < word.length(); i++) {
            index = word.charAt(i) - 'a';
            if (node.nexts[index] == null) {
                return false;
            }
            // 指针移动
            node = node.nexts[index];
        }
        return node.isEnd;
    }
    /** Returns if there is any word in the trie that starts with the given prefix. */
    public boolean startsWith(String prefix) {
        if (prefix == null) {
            return false;
        }
        TrieNode node = root;
        int index = 0;
        for (int i = 0; i < prefix.length(); i++) {
            index = prefix.charAt(i) - 'a';
            if (node.nexts[index] == null) {
                return false;
            }
            node = node.nexts[index];
        }
        return true;
    }
}
/**
 * Your Trie object will be instantiated and called as such:
 * Trie obj = new Trie();
 * obj.insert(word);
 * boolean param_2 = obj.search(word);
 * boolean param_3 = obj.startsWith(prefix);
 */


T421 数组最大异或值


待补充。。。

相关文章
|
存储 Java
Java数据结构之第十五章、Trie(前缀树/单词查找树)
1.前缀树的概念:前缀树又叫字典树或单词查找树(高效的存储和查找字符串集合的数据结构)。2.3.存储形式:存储的字符串可能:全是 小写字母 或全是 大写字母 或全是 数字 或全是 0和1。它是一棵,每个代表一个,从。字典树的根节点不包含字符,每个子节点代表一个字符,从根节点到任意一个节点所经过的路径上的字符连接起来即为该节点所代表的字符串。每个节点可以存储一个或多个字符串,通常使用一个标志来标记一个节点代表的字符串是否存在。当需要在一组字符串中查找某个字符串时,可以利用字典树来实现高效的查找操作。
75 0
|
5天前
|
C语言
【数据结构】栈和队列(c语言实现)(附源码)
本文介绍了栈和队列两种数据结构。栈是一种只能在一端进行插入和删除操作的线性表,遵循“先进后出”原则;队列则在一端插入、另一端删除,遵循“先进先出”原则。文章详细讲解了栈和队列的结构定义、方法声明及实现,并提供了完整的代码示例。栈和队列在实际应用中非常广泛,如二叉树的层序遍历和快速排序的非递归实现等。
64 9
|
2天前
|
存储 JavaScript 前端开发
执行上下文和执行栈
执行上下文是JavaScript运行代码时的环境,每个执行上下文都有自己的变量对象、作用域链和this值。执行栈用于管理函数调用,每当调用一个函数,就会在栈中添加一个新的执行上下文。
|
4天前
|
存储
系统调用处理程序在内核栈中保存了哪些上下文信息?
【10月更文挑战第29天】系统调用处理程序在内核栈中保存的这些上下文信息对于保证系统调用的正确执行和用户程序的正常恢复至关重要。通过准确地保存和恢复这些信息,操作系统能够实现用户模式和内核模式之间的无缝切换,为用户程序提供稳定、可靠的系统服务。
26 4
|
27天前
|
算法 程序员 索引
数据结构与算法学习七:栈、数组模拟栈、单链表模拟栈、栈应用实例 实现 综合计算器
栈的基本概念、应用场景以及如何使用数组和单链表模拟栈,并展示了如何利用栈和中缀表达式实现一个综合计算器。
26 1
数据结构与算法学习七:栈、数组模拟栈、单链表模拟栈、栈应用实例 实现 综合计算器
|
8天前
|
算法 安全 NoSQL
2024重生之回溯数据结构与算法系列学习之栈和队列精题汇总(10)【无论是王道考研人还是IKUN都能包会的;不然别给我家鸽鸽丢脸好嘛?】
数据结构王道第3章之IKUN和I原达人之数据结构与算法系列学习栈与队列精题详解、数据结构、C++、排序算法、java、动态规划你个小黑子;这都学不会;能不能不要给我家鸽鸽丢脸啊~除了会黑我家鸽鸽还会干嘛?!!!
|
27天前
初步认识栈和队列
初步认识栈和队列
57 10
|
21天前
数据结构(栈与列队)
数据结构(栈与列队)
16 1
|
28天前
|
算法
数据结构与算法二:栈、前缀、中缀、后缀表达式、中缀表达式转换为后缀表达式
这篇文章讲解了栈的基本概念及其应用,并详细介绍了中缀表达式转换为后缀表达式的算法和实现步骤。
41 3
|
26天前
|
存储 JavaScript 前端开发
为什么基础数据类型存放在栈中,而引用数据类型存放在堆中?
为什么基础数据类型存放在栈中,而引用数据类型存放在堆中?
61 1