基础算法 835.Trie字符串

#include<iostream>
using namespace std;

const int N = 100010;
int son[N][26], cnt[N], idx;
char str[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(){
    int n;
    char d[2];
    scanf("%d",&n);
    while(n--){
        scanf("%s%s",d,str);
        if(*d == 'I') insert(str);
        else printf("%d\n",query(str));
    }

    return 0;
}