2020.01.31 七分糖 Trie 字符串统计

字符串统计（Trie）

可以统计字符串在集合中出现的次数

模板：
ACWing 835
题意：
I ： 在集合中插入字符串
Q ：查询字符串在集合中出现次数

#include 
#include 

using namespace std;

const int N = 100010;

int son[N][26], cnt[N], idx; //son[N][26]:每个节点连接的字母 || cnt[N]:统计字符串出现次数
char op[2], ch[N];
int n;

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];
}

int main(){
    scanf("%d",&n);
    while (n --){
        scanf("%s%s",&op,&ch);
        if (op[0] == 'I') insert(ch);
        else printf("%d\n",query(ch));
    }
    return 0;
}