C++ 数组实现 Trie 树

int son[N][26], cnt[N], idx;  // son[][] 和 cnt[] 以及 idx 前提都是置 0 的。
// 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;  // if (!son[p][u]) 代表的意思是:在结点 p 的后代中不存在 u 结点, son[p][u] = ++ idx 代表的意思是 不存在就创建一个后代。
        p = son[p][u];  // 沿着Trie树向下走
    }
    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];
}