数据结构——trie树（字典树）

#include<iostream>
using namespace std;

const int N = 200010;
//用来存放子节点，idx是已经使用的结点下标
int son[N][26],idx;
//用于标记是否有单词
int cnt[N];

void insert(char str[]){
    //p == 0即是根节点，也是空节点
    int p = 0;
    for(int i = 0;str[i];i++){
        int t = str[i] - ‘a‘;
        //判断是否有当前字符的子节点
        if(!son[p][t]) son[p][t] = ++ idx;
        
        p = son[p][t];
    }
    //标记字符串末尾
    cnt[p]++;
}

int query(char str[]){
    int p = 0;
    for(int i = 0;str[i]; i ++){
        int t = str[i] - ‘a‘;
        if(!son[p][t]) return 0;
        p = son[p][t];
    }
    if(cnt[p]) return cnt[p];
}

int main(){
    int n;
    cin>>n;
    char ch;
    char str[N];
    for(int i = 0 ; i < n ;i ++){
        cin>>ch>>str;
        if(ch == ‘I‘){
            insert(str);
        }
        if(ch == ‘Q‘){
            cout<<query(str)<<endl;
        }
    }
    
    
    
    return 0;
}