Trie树——Trie字符串统计

Tire：Tire树又称字典树或单词查找树，是一种高效的存储和查找字符串集合的数据结构

例题：

#include <iostream>

using namespace std;

const int N = 100010;

//数组son表示当前节点的孩子
//例如son[1][0]=2表示1结点的一个值为a的子结点为结点2，son[1][0] = 0，则意味着没有值为a子结点
//这里的son[N][26]相当于链表中的ne[N]
//数组cnt表示叶子节点被用到的次数
//idx就相当于一个分配器
int son[N][26], cnt[N], idx;
char str[N];

void insert(char* str)
{
    int p = 0;//类似指针的功能，从很节点开始遍历,根节点刚开始赋值为0
    for(int i = 0; str[i]; ++i)
    {
        int u = str[i] - 'a';//26个字母映射为0-25
        if(!son[p][u]) son[p][u] = ++idx;//没有该子节点就创建一个
        p = son[p][u];//走到p的子节点
    }
    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 == 'I') insert(str);
        else printf("%d\n", query(str));
    }
    
    return 0;
}