字典树字典树,一个字符串就是一个树结构
下面代码中,
idx是每一个字符串都会有的独一无二的一个序号,
第一个加进来的字符串的序号为1,后面加进来的序号按照进来的顺序去+1
。
然后因为你用数组模拟树嘛,你的数组是son[p][u]
p代表你已经走到哪个字符串的位置了(哪个树结构的位置)?
所以一开始是0,代表都没有走到。
对于树的理解也加深了,就你每走到一步,或者说你一步走完,你就要变换一下起点,以这一步为起点进行你的下一轮征程。
#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 ++ )
{
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 n;
scanf("%d", &n);
while (n -- )
{
char op[2];
scanf("%s%s", op, str);
if (*op == 'I') insert(str);
else printf("%d\n", query(str));
}
return 0;
}