维护一个字符串集合,支持两种操作:
I x 向集合中插入一个字符串 x; Q x 询问一个字符串在集合中出现了多少次。 共有 N 个操作,所有输入的字符串总长度不超过
105,字符串仅包含小写英文字母。输入格式 第一行包含整数 N,表示操作数。
接下来 N 行,每行包含一个操作指令,指令为 I x 或 Q x 中的一种。
输出格式 对于每个询问指令 Q x,都要输出一个整数作为结果,表示 x 在集合中出现的次数。
每个结果占一行。
数据范围
1≤N≤2∗10000
输入样例:
5
I abc
Q abc
Q ab
I ab
Q ab
输出样例:
1
0
1
分析
int son[N][26], cnt[N], idx;
就是一个就是一个邻接表
son[N][26]就是分配的N个节点给trie使用,同时每个节点有26个状态,每次只有一个状态是成立的,而表头就是son[0][26]每个状态对应一个开头
而insert操作就是一个尾插法,有头节点的链表尾插就要遍历到最后一个元素。这个for循环就是在做这个操作
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]; //走到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];
}
就是一次邻接表的遍历过程,如果路径不通就返回0
代码
#include<bits/stdc++.h>
#include<algorithm>
using namespace std;
const int N = 100010;
int son[N][26],cnt[N];
int n, p, idx;
//插入
void insert(char str[]){
p = 0;//目前所在的节点
//遍历字符串里面的字符
for(int i = 0; str[i]; i++){
//26个字母,0~25
int u = str[i] - 'a';
//以abc为例
//P等于0,如果0的子节点,a不存在,idx++,idx 就相当于存储空间的下标
if(!son[p][u]) son[p][u] = ++idx;
p = son[p][u];//告诉p你下一步往哪里走
}
cnt[p] ++;//这个字符串遍历完,记住他插入的次数
}
//查询
int query(char str[]){
p = 0;
for(int i = 0; str[i]; i++){
//u的作用,u代表下一个字母,是唯一的,这也使得p是唯一的
int u = str[i] - 'a';
//如果这个字母的下一个字母不存在,直接返回
if(!son[p][u]) return 0;
p = son[p][u];
}
return cnt[p];//返回重复次数
}
int main(){
cin >> n;
char c;
char x[N];
while(n--){
cin >> c;
if(c == 'I'){
cin >> x;
insert(x);
}else if(c == 'Q'){
cin >> x;
printf("%d \n",query(x));
}
}
return 0;
}