Trie树笔记(1)

文章介绍了Trie树作为一种高效存储和查找字符串集合的数据结构,主要阐述了如何进行插入和查询操作。在插入时,按照字符串逐字符创建节点;查询时,根据路径判断是否存在单词。代码示例展示了C++实现的插入和查询函数。
摘要由CSDN通过智能技术生成

这是一个非常非常简单的数据结构,正好适合用来作为学完KMP的一个缓冲。

Trie树是一个用来高效地存储和查找字符串集合的数据结构,在算法竞赛中,只会规定很少的结点种类,比如所有的小写英文字母,或者所有大写英文字母等等。

先以存一个abcdef为例,trie树有一个根节点root,接下来我们查询是否有a子结点,如果没有,那就创建一个,以此类推,我们一节一节创建结点,直到把f存完。

如果接下来要存一个bdef,那就会从之前创建的b结点往后存。

在存完所有词之后,我们给所有末端打上标记

当我们查找的时候,我们就看有没有标记就行了。如果发现走到一半没有路径了或者到了位置却没有标记,那就说明我们没有存入这个单词。

具体用法背出一道模板题即可:

代码如下:

#include<iostream>
using namespace std;

const int N = 100010;

int son[N][26];//son子结点,由于题目只要求小写英文字母,那么每一个结点的子结点最多有26个,N表示结点的总数
int cnt[N];//以当前这个点结尾的单词有多少个(重复单词)
int idx;//当前用到的是哪个下标,下标是0 的点,即是根节点,又是空节点
char str[N];

//插入操作
void insert (char str[]) {
    int p = 0;
    for (int i = 0; str[i]; i ++) {
        int u = str[i] - 'a';//把当前这个字母对应的子结点的编号映射为0~25的数
        if (!son[p][u]) son[p][u] = ++ idx;
        p = son[p][u];
    }
    cnt[p] ++;
}

//查询操作
int query (char str[]) {
    int p = 0;
    for (int i = 0; str[i]; i ++) {
        int u = str[i] - 'a';
        if (!son[p][u]) return 0;
        p = son[p][u];
    }
    return cnt[p];
}


int main () {
    int n;
    scanf("%d", &n);
    while (n --) {
        char op[2];
        scanf("%s%s", op, str);
        if (op[0] == 'I') insert(str);
        else printf("%d\n", query(str));
    }
    
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值