:::info 参考博文:https://www.acwing.com/solution/content/14695/
课程链接:https://www.acwing.com/video/260/
例题链接:https://www.acwing.com/problem/content/837/ :::

思想

用于高效的存储和查找字符串集合的数据结构,用Trie树的字符串一般要么是小写字母要么都是大写字母,类型不会多,字符串的长度也不会太长
root为空节点,图中五角星的意思是标记有以该字母结尾的字符串,代码表示是cnt数组,记录以该字母结尾的字符串的数量
28CAEECBA904DC6239B22E253914A55A.png

例题

image.png
该题目中都为26个字母,所以可以开辟son[][26]数组,来映射26个字母,该数组表示的是子节点,用son[]某节点的所有子节点,son[]列映射为子节点的值,son[][]的值表示该节点是第几个加入的不重复的节点。cnt[]数组标识字符串的结束,idx用来确定节点是否存在
image.png

代码

  1. #include <iostream>
  2. using namespace std;
  3. const int N = 100010;
  4. int son[N][26], cnt[N], idx;
  5. char str[N];
  6. void insert(char *str)
  7. {
  8. int p = 0;
  9. for (int i = 0; str[i]; i ++ )
  10. {
  11. int u = str[i] - 'a';
  12. if (!son[p][u]) son[p][u] = ++ idx;
  13. p = son[p][u];
  14. }
  15. cnt[p] ++ ;
  16. }
  17. int query(char *str)
  18. {
  19. int p = 0;
  20. for (int i = 0; str[i]; i ++ )
  21. {
  22. int u = str[i] - 'a';
  23. if (!son[p][u]) return 0;
  24. p = son[p][u];
  25. }
  26. return cnt[p];
  27. }
  28. int main()
  29. {
  30. int n;
  31. scanf("%d", &n);
  32. while (n -- )
  33. {
  34. // 此处设置成op[2],而不是char op,是因为char op用scanf("%c")会读入一些空格或回车
  35. // 而char op[2]用scanf("%s")会忽略空格和回车
  36. char op[2];
  37. scanf("%s%s", op, str);
  38. if (*op == 'I') insert(str);
  39. else printf("%d\n", query(str));
  40. }
  41. return 0;
  42. }