Trie树 数组表示

感谢CSDN大佬

#include <bits/stdc++.h>
using namespace  std;
const int MAX=1000010;
int trie[MAX][26];
int val[MAX];
int tot;
void add(string s)
{
        int t=0;
        for(int i=0;i<s.size();i++)
        {
            int c=s[i]-'a';
            if(trie[t][c]==0)trie[t][c]=++tot;
            t=trie[t][c];
        }
        val[t]++;
}
int find(string s)
{
    int t=0;
    for(int i=0;i<s.size();i++)
    {
        int c=s[i]-'a';
        if(trie[t][c]==0)return 0;//一个也没有
        t=trie[t][c];
    }
    return val[t];//返回有几个这样的字符
}
using namespace  std;
int main()
{
    //trie树的数组表示
    int N;
    char c;
    cin>>N;
    string s;
    for(int i=0;i<N;i++)
    {
        cin>>c>>s;
        if(c=='I')add(s);
        else if(c=='Q')cout<<find(s)<<endl;
    }
}