Trie字符串统计

Trie树又称字典树、单词查找树。是一种能够高效存储和查找字符串集合的数据结构。

代码中，我们用到了son[N][26],cnt[N],idx;

我们将26个英文字母转换成数0-26，N代表当前节点，son[N][]代表当前节点的儿子。如果没有的话，可以等于++idx。Trie树本质上是一颗多叉树，对于字母而言最多有26个子结点。所以这个数组包含了两条信息。比如：son[1][0]=2表示1结点的一个值为a的子结点为结点2;如果son[1][0] = 0，则意味着没有值为a子结点。这里的son[N][26]相当于链表中的ne[N]。

为了能够唯一标识化每个节点，需要让每个节点唯一不重复，因此这里用到了idx；

cnt[N]是用来存储以N为结尾的字符串的个数。

insert操作就是挨个字母查找son[][]是否为0，为0表示没有这个节点，所以就新建节点，给这个节点一个idx值。若不为0，就继续找儿子节点，直到字符串循环完毕。若全都存在，则以末尾字母为节点的个数加一。

#include<iostream>
#include<string>
using namespace std;
const int N=100010;
int son[N][26],cnt[N],idx;
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];
	}
	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 m;
	cin>>m;
	while(m--){
		char op[2];
		scanf("%s%s",op,str);
		if(*op=='I') insert(str);
		else printf("%d\n",query(str));
	}
	return 0;
	
}