Trie树的应用
- 题目
- 解题思路
- 代码
题目
维护一个字符串集合,支持两种操作:
I x
向集合中插入一个字符串 x x x;Q x
询问一个字符串在集合中出现了多少次。
共有 N N N 个操作,所有输入的字符串总长度不超过 1 0 5 10^5 105,字符串仅包含小写英文字母。
输入格式
第一行包含整数 N N N,表示操作数。
接下来 N N N 行,每行包含一个操作指令,指令为 I x
或 Q x
中的一种。
输出格式
对于每个询问指令 Q x
,都要输出一个整数作为结果,表示 x x x 在集合中出现的次数。
每个结果占一行。
数据范围
KaTeX parse error: Undefined control sequence: \* at position 14: 1 \le N \le 2\̲*̲10^4
输入样例:
5
I abc
Q abc
Q ab
I ab
Q ab
输出样例:
1
0
1
解题思路
这个题需要我们创建一个trie树来解决问题。
假设我们需要插入的两个字符串是 abc 和 abdef,此时树就长这个样子。
假设再插入一个 abdf,那么就变成这个样子
本题还需要能够查询 一个字符串中出现的次数
所以在每次插入一个字符串时,都需要在末尾的字母处标记,具体怎么操作的还需要结合一下代码
代码
首先准备阶段如下:
其中 数组 son每一行代表 一个节点,每一行当中的列代表着 他是 a b c … z 中的哪一种字母,它存储的值,是他的儿子,也就是它的下一个节点的位置,如果值是0的话则代表它没有儿子。
idx代表着现在数组 son 用到那个点了。
数组cnt 用来标记每一个字符串出现了多少次。
数组str 用来临时存储要插入的字符串。
插入函数:
参数为一个字符串,写成数组和 指针的形式都可以。
这个p的含义是 数组son 的行数,也就是表示节点的位置。
p = 0代表当前 为 根节点。
接下来遍历整个字符串,拿到字符串的每一个字母所对应的每一行的列坐标(比如 a对应 0下标,b对应 1下标)
这句话的意思是,如果此时没有该处没有被插入值也就是为0时,那么此时就创建一条路,给他个儿子节点,这个儿子的位置由 idx 决定,Idx的一生会从 1 开始 加加,作为每次插入点的坐标。
接着 p 更新到它儿子节点的位置。
最后在 数组cnt p位置上 自增1。
至此插入操作就完成了。
接着写查询函数。
同样,与插入的前面部分相同,u 是字符串中字母所对应的下标
与前面不同的点是,插入操作是 没路创造路,而查询操作是 没路则代表他这个字符串就不存在,所以就出现了0次,所以返回0.
接着还是 指到下一个节点
最后遍历完之后,如果没有被 return 0,那么此时 cnt[ p ] 的值就是 该字符串出现的 次数。
接着是main函数部分
main函数我们只需要根据题目当中的输入和输出,然后调用对应的方法即可。
完整代码:
#include <iostream>
using namespace std;const int N = 1e5+10;int son[N][26], idx, cnt[N];
char str[N];int n;void insert(char* str)
{int p = 0;for (int i = 0; str[i] != 0; i++){int u = str[i] - 'a';if(son[p][u] == 0) 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] == 0) return 0;p = son[p][u];}return cnt[p];
}int main()
{scanf("%d", &n);while (n -- ){char op[2];scanf("%s", op);scanf("%s", str);if (op[0] == 'I') insert(str); else printf("%d\n", query(str));}return 0;
}
完