AcWing835 Trie字符串统计

#include<iostream>
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++) {//这里判断可以偷个小懒就是str最后一位是'/0'可以直接拿来判断结束
		int u = str[i] - 'a';
		if (!son[p][u])son[p][u]=++idx;//p表示当前节点的大概位置 
		//u表示字母 二者共同确定位置 idx++指向下一个节点的大概位置
		//如果没有路径那就创建一个新的路径
		//p大概可以理解成第几层？ 
		//u为某一层的具体点的位置？ 二者共同确定该字符的位置？
		//注意 idx为全局变量 故一直递增 避免出现定位不明的情况
		p = son[p][u];//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 op[2];
		scanf("%s%s", &op, &str);

		if (*op == 'I') insert(str);
		else printf("%d\n", query(str));
	}

	return 0;
}