Trie树/字典树

Trie树/字典树

作用

用于高效的存储和查找字符串集合的一个数据结构

食用范围

字符串题，要么全是小写字母，要么全是大写字母，要么全是数字，字母的类型不多，不光可以存储字符串，也可以存储整数、二进制数。

Code

int son[N][26], cnt[N], idx; //cnt存储以p结尾的单词的个数
// 下标为0的点，既是根结点，又是空结点
char str[N];

void insert(char s[]) {
    int p = 0;
    for (int i = 0; s[i]; i++) {
        int u = s[i] - 'a';
        if (!son[p][u]) son[p][u] = ++idx;
        p = son[p][u];
    }

    cnt[p]++;

}

int query(char s[]) {
    int p = 0; //²ãÊý
    for (int i = 0; s[i]; i++) {
        int u = s[i] - 'a';
        if (!son[p][u]) return 0;
        p = son[p][u];
    }

    return cnt[p];
}