[字典树+模板] Trie树模板

it2025-08-03  7

文章目录

0. 前言1. Trie

0. 前言

Biu

Trie:高效地存储和查找字符串集合的数据结构,也称字典树、前缀树。

数据一定限制了字母的种类,大多都是小写字母、大写字母等等,一般都是 26 个,或者 56 个。

1. Trie

这里的 son[N][26] 相当于链表中的 ne[N],存放的是下一个节点的位置,通过 idx++ 的操作,保证了不同的 idx 值对应不同的节点。这个和数组模拟单链表、双链表的思路一致,我的之前博文并没有总结,后续做图论、模拟堆等还会大量遇见,需要转变下思维,理解这种数据存储方式。

模板代码:

#include <iostream> using namespace std; const int N = 1e5+5; int m; char str[N]; int son[N][26], cnt[N], idx; // 根节点和空节点的下标均为0,idx类似于单链表 void insert(char str[]) { int p = 0; for (int i = 0; str[i]; ++i) { // c++字符串结尾\0,可以以此判断是否结尾 int u = str[i] - 'a'; if (!son[p][u]) son[p][u] = ++idx; // 不存在则创建,前置++,树头结点为0,存储从1开始,保存下一个节点的位置 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() { cin >> m; while (m --) { char op[2]; cin >> op >> str; if (op[0] == 'I') insert(str); else cout << query(str) << endl; } return 0; }
最新回复(0)