【AcWing】835. Trie字符串操作(Trie树)

Trie树的概念和构建方法

Trie 树（也称为前缀树或字典树）是一种用于快速检索字符串数据集中的键的树形数据结构。

Trie: 高效地存储和查找字符串集合的数据结构。

使用Trie树存储abcdef、abdef、aced、bcdf、bcff、cdaa、bcdc和abc这些字符串。

建树规则：对每一个字符串，从根节点向下看，如果存在与字符串相同的字符就往下走，如果没有相同的字符就在当前位置增加新的字符子节点，直到字符串完结，在对应节点打一个标记“ * ”，用于表示以当前节点结束存在一个字符串。(0~23为编号。)

三种查询情况：
1.查找到：查询aced时，有1-10-11-12-*这条路径，所以存在aced字符串。

2.字符串不匹配：查询abcf，走1-2-3-之后不存在字符f，所以不存在abcf字符串。

3.字符串匹配但不存在：查询abcd，走1-2-3-4但是没有*号标记，所以不存在abcd字符串。

题目

 

题解代码 

#include <iostream>

using namespace std;

const int N = 100010;

int son[N][26]/*小写字母只有26个，每一个位置对应一个小写字母*/;
int cnt[N]/*存储以此结尾的字符串数*/;
int idx/*当前位置编号*/;
char str[N];
//idx=0的节点是根节点，也是空节点

// 插入一个字符串
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;//不存在就创建一个位置并写上编号，
//通编号可以查找到以这个字符为根的树，每个点都可以有一个树，一层分支最多26个
        p = son[p][u];
    }
    cnt[p] ++ ;//以此结尾的字符串数+1
}

//查询字符串出现的次数
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 == 'I') insert(str);
        else printf("%d\n", query(str));
    }

    return 0;
}