10.9

Trie树
维护一个字符串集合，支持两种操作：
I x 向集合中插入一个字符串 x；
Q x 询问一个字符串在集合中出现了多少次。
共有 N个操作，所有输入的字符串总长度不超过 105，字符串仅包含小写英文字母。

输入格式
第一行包含整数 N，表示操作数。

接下来 N行，每行包含一个操作指令，指令为 I x 或 Q x 中的一种。

输出格式
对于每个询问指令 Q x，都要输出一个整数作为结果，表示 x在集合中出现的次数。
每个结果占一行。

数据范围
1≤N≤2∗104

输入样例：
5
I abc
Q abc
Q ab
I ab
Q ab
输出样例：
1
0
1

include

using namespace std;
const int N=1000010;
int son[N][26],cnt[N],idx;
char str[N];
//插入 如果每次是新节点时son[][]新建一个节点，即idx++即可，创建节点，并将指针指向下一个位置即p=son[p][u]
void insert(string s)
{
int p=0;
for(int i=0;i<s.size();i++)
{
int u=s[i]-'a';//字母转化为数字
if(!son[p][u])son[p][u]=++idx;
p=son[p][u];
}
cnt[p]++;//结束时的标记，也是记录以此节点结束的字符串个数
}

int query(string s)
{
int p=0;
for(int i=0;i<s.size();i++)
{
int u=s[i]-'a';
if(!son[p][u])return 0;//节点不存在，返回0即可，代表该字符串不存在
p=son[p][u];
}
return cnt[p];//cnt[]代表字符串出现的次数
}
int main()
{
int n;
char c;
cin>>n;
while(n--)
{
cin>>c;
if(c'I')
{
string s;
cin>>s;
insert(s);
}
if(c'Q')
{
string s;
cin>>s;
cout<<query(s)<<endl;
}
}
}