前缀树

1.什么是前缀树

前缀树是N叉树的一种特殊形式。通常来说,一个前缀树是用来存储字符串的。前缀树的每一个节点代表一个字符串(前缀)。每一个节点会有多个子节点,通往不同子节点的路径上有着不同的字符。子节点代表的字符串是由节点本身的原始字符串,以及通往该子节点路径上所有的字符组成的。

下面是前缀树的一个例子


image.png

在上图示例中,我们在节点中标记的值是该节点对应表示的字符串。例如,我们从根节点开始,选择第二条路径 'b',然后选择它的第一个子节点 'a',接下来继续选择子节点 'd',我们最终会到达叶节点 "bad"。节点的值是由从根节点开始,与其经过的路径中的字符按顺序形成的。

值得注意的是,根节点表示空字符串。

前缀树的一个重要的特性是,节点所有的后代都与该节点相关的字符串有着共同的前缀。这就是前缀树名称的由来。

我们再来看这个例子。例如,以节点 "b" 为根的子树中的节点表示的字符串,都具有共同的前缀 "b"。反之亦然,具有公共前缀 "b" 的字符串,全部位于以 "b" 为根的子树中,并且具有不同前缀的字符串来自不同的分支。

前缀树有着广泛的应用,例如自动补全,拼写检查等等。我们将在后面的章节中介绍实际应用场景。

2 如何表示一个前缀树?

前缀树的特别之处在于字符和子节点之间的对应关系。有许多不同的表示前缀树节点的方法,这里我们只介绍其中的两种方法。

2.1 数组

第一种方法是用数组存储子节点。

例如,如果我们只存储含有字母 a 到 z 的字符串,我们可以在每个节点中声明一个大小为26的数组来存储其子节点。对于特定字符 c,我们可以使用 c - 'a' 作为索引来查找数组中相应的子节点。

// change this value to adapt to different cases
#define N 26

struct TrieNode {
    TrieNode* children[N];
    
    // you might need some extra values according to different cases
};

/** Usage:
 *  Initialization: TrieNode root = new TrieNode();
 *  Return a specific child node with char c: (root->children)[c - 'a']
 */

访问子节点十分快捷。访问一个特定的子节点比较容易,因为在大多数情况下,我们很容易将一个字符转换为索引。但并非所有的子节点都需要这样的操作,所以这可能会导致空间的浪费。

#include <iostream>
#include <map>
#include <vector>
#define N 26
using namespace std;
class Trie{
    private:
        struct Node{
            Node* next[N];
            bool is_w = false;//是否是一个完整的单词
        };
        Node* trie;
    public:
        Trie(){
            trie = new Node();
        }
        /*insert word*/
        void insert(const string & str){
            Node* cur = trie;
            for(int i = 0;i < str.size();i++){
                if(cur->next[str[i] - 'a'] == nullptr)
                    cur->next[str[i] - 'a']= new Node();
                cur = cur->next[str[i] - 'a'];
            }
            cur->is_w = true;
        }
        bool search(const string & str) {
            Node* cur = trie;
            for(int i = 0;i < str.size();i++){
                if(cur->next[str[i] - 'a'] != nullptr)
                    cur = cur->next[str[i] - 'a'];
                else
                    return false;
            
            }
            if(cur->is_w) return true;
            else return false;
        }
};

int main(){
    Trie t;
    t.insert("apple");
    t.insert("banana");
    cout << t.search("apple") << endl;
    cout << t.search("banana") << endl;
    return 0;
}

2.2 Map

第二种方法是使用 Hashmap 来存储子节点。
我们可以在每个节点中声明一个Hashmap。Hashmap的键是字符,值是相对应的子节点。

struct TrieNode {
    unordered_map<char, TrieNode*> children;
    
    // you might need some extra values according to different cases
};

/** Usage:
 *  Initialization: TrieNode root = new TrieNode();
 *  Return a specific child node with char c: (root->children)[c]
 */

通过相应的字符来访问特定的子节点更为容易。但它可能比使用数组稍慢一些。但是,由于我们只存储我们需要的子节点,因此节省了空间。这个方法也更加灵活,因为我们不受到固定长度和固定范围的限制。

我们已经提到过如何表示前缀树中的子节点。除此之外,我们也需要用到一些其他的值。

例如,我们知道,前缀树的每个节点表示一个字符串,但并不是所有由前缀树表示的字符串都是有意义的。如果我们只想在前缀树中存储单词,那么我们可能需要在每个节点中声明一个布尔值(Boolean)作为标志,来表明该节点所表示的字符串是否为一个单词。

3 leetcode题目最长公共前缀

思路非常简单,由于是寻找最长前缀,所以只需要插入第一个字符串,后续字符串分别进行匹配即可,更优化一点可以先遍历字符数组,找到最短的那个字符串,再进行匹配。

class Solution {
public:
    struct Node{
        Node* next[26];
        bool isStr;
    };
    void insertTrie(Node* root,const string &str){
        if(root == nullptr) return;
        for(int i = 0;i < str.size();i++){
                if(root->next[str[i] - 'a'] == nullptr){
                    root->next[str[i] - 'a'] = new Node();
                }
                root = root->next[str[i] - 'a'];
        }
        root->isStr = true;
    }
    int searchCommonPrefix(Node* root, const string& str){
        if(root == nullptr) return 0;
        int maxCommon = 0;
        for(int i = 0;i < str.size();i++){
            if(root->next[str[i] - 'a'] == nullptr)
                return maxCommon;
            else{
                maxCommon++;
                root = root->next[str[i] - 'a'];
            }
        }
        return maxCommon;
    }
    string longestCommonPrefix(vector<string>& strs) {
        string result = "";
        if(strs.size() == 0) return result;
        if(strs.size() == 1) return strs[0];
        Node* trie = new Node();
        insertTrie(trie,strs[0]);
        int maxCommon = strs[0].size();
        for (int i = 1;i < strs.size();i++){
            maxCommon = maxCommon > searchCommonPrefix(trie,strs[i]) ? searchCommonPrefix(trie,strs[i]) : maxCommon;
            if(maxCommon == 0) return result;
        }

        for(int i = 0;i <maxCommon;i++){
            result += strs[0][i];
        }
        return result;
    }
};

字典树参考:https://www.cnblogs.com/vincent1997/p/11237389.html

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 212,080评论 6 493
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,422评论 3 385
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 157,630评论 0 348
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,554评论 1 284
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,662评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,856评论 1 290
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,014评论 3 408
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,752评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,212评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,541评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,687评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,347评论 4 331
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,973评论 3 315
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,777评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,006评论 1 266
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,406评论 2 360
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,576评论 2 349

推荐阅读更多精彩内容

  • 本文只是自己的笔记,并不具备过多的指导意义。 前缀树 何为前缀树 前缀树又名字典树,单词查找树,Trie树,是一种...
    kirito_song阅读 4,183评论 0 13
  • 引子 在刷题的过程中,经常会遇到这样一种典型问题: 给一组字符串List strs,找出其中前缀为String ...
    akak18183阅读 209评论 0 0
  • 叫前缀树更容易理解字典树的样子 Trie又被称为前缀树、字典树,所以当然是一棵树。上面这棵Trie树包含的字符串集...
    Drama_Du阅读 400评论 0 0
  • 前缀树是什么 前缀树是一种树结构,其中的键通常是字符串。与二叉查找树不同,键不是直接保存在节点中,而是由节点在树中...
    名字是乱打的阅读 595评论 0 1
  • 什么是Trie? Trie树,也叫作字典树或前缀树,顾名思义,它是一个树行结构。它是一种专门处理字符串匹配的数据结...
    花逝97阅读 619评论 0 2