Tire字符串统计（Acwing算法基础课）

Tire:高效的存储和查找字符串集合的数据结构

题目来源：Acwing---835

#include<iostream>
using namespace std;
const int N=100010;
int son[N][26],cnt[N],idx;//son[N][26]:每个节点都有26个孩子		idx:下标是0的点，既是根节点，又是空结点
char str[N];
void insert(char str[]){
	int p=0;
	for(int i=0;str[i];i++){
		int u=str[i]-'a';//映射为0~25
		if(!son[p][u])son[p][u]=++idx;//没有这个单词就把这个单词加进来
		p=son[p][u];
	}
	cnt[p]++;//以(p+'a')结束的单词个数
}
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;
	scanf("%d",&n);
	while(n--){
		char op[2];
		scanf("%s%s",op,str);
		if(op[0]=='I')insert(str);
		else printf("%d\n",query(str));
	}
	return 0;
}