Trie 树

邪童的博客 / 2023-05-03 / 原文

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


//Trie树模板
int son[N][26],cnt[N],idx;

//插入一个字符串
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];
}