作用
高效存储和查找字符串集合的数据结构
图解
实例图
对应的结果图
模板题-AcWing
Trie字符串统计
维护一个字符串集合,支持两种操作:
I x
向集合中插入一个字符串 x;Q x
询问一个字符串在集合中出现了多少次。共有 N 个操作,输入的字符串总长度不超过 10^5,字符串仅包含小写英文字母。
输入格式
第一行包含整数 N,表示操作数。
接下来 N 行,每行包含一个操作指令,指令为
I x
或Q x
中的一种。
输出格式
对于每个询问指令
Q x
,都要输出一个整数作为结果,表示 x 在集合中出现的次数。每个结果占一行。
数据范围
1≤N≤2∗10^4
输入样例:
5
I abc
Q abc
Q ab
I ab
Q ab
输出样例:
1
0
1
代码
#include <iostream>
using namespace std;
const int N = 100010;
//该节点的儿子节点,以该节点结尾的数量,当前用到的节点数
int son[N][26],cnt[N],idx;//下标为0的节点,既是根节点也是空节点
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];//顺这儿子节点向下存储
// printf("%d------\n",p);
}
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;
cin>>n;
while(n--){
char c;
cin>>c>>str;
if(c=='I')insert(str);
else cout<<query(str)<<endl;
}
}