【ACWing 算法基础】Trie（字典树）

一. 模板

• 用数组模拟 Trie 树

int son[N][26], cnt[N], idx;
// 0号点既是根节点，又是空节点
// son[][]存储树中每个节点的子节点
// cnt[]存储以每个节点结尾的单词数量

// 插入一个字符串
void insert(char *str)
{
    int p = 0;
    for (int i = 0; str[i]; i ++ )
    {
        int u = str[i] - 'a';
        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];
}

二. 总结

• 如何理解单(双)链表，Trie树和堆中的idx？
• Trie 树结构如下：
  

三. 例题

• 835. Trie字符串统计

AC代码:

#include <iostream>

using namespace std;

const int N = 1e5 + 10;

int son[N][26], cnt[N], idx;  // 下标0代表根节点和空节点，cnt用于计数，idx代表结点的索引
char str[N];

void insert(char* str)
{
    int p = 0;  // 根节点
    for (int i = 0; str[i]; ++i) {
        int u = str[i] - 'a';  // 映射字母a-z为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';  // 映射字母a-z为0-25
        if (!son[p][u]) return 0;  // 没有该子结点就返回查询不到
        p = son[p][u];  // 走到该子节点
    }
    
    return cnt[p];  // 返回该子节点存在的单词个数    
}

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    int n;
    cin >> n;
    char op;
    while (n--) {
        cin >> op >> str;
        if (op == 'I')  insert(str);
        else cout << query(str) << endl;
    }
    return 0;
}