字典树,也称为前缀树(Trie树),是一种用于高效存储和检索字符串数据的树形数据结构。它是一种多叉树,通常用于实现字符串的快速检索,例如自动完成、拼写检查和字符串搜索等应用。
字典树的基本特点如下:
1. **节点结构**:字典树的每个节点包含若干个指向子节点的指针,通常使用数组或哈希表来实现。每个节点还可能包含一个标识字段,用于表示该节点对应的字符是否构成一个完整的单词。
2. **前缀匹配**:字典树的主要优势在于能够高效地进行前缀匹配。通过从根节点开始,依次沿着字符串的字符构建路径,直到找到目标字符串或者无法继续匹配为止。
3. **高效性**:字典树能够在时间复杂度为 O(m) 的情况下,进行字符串的插入、查找和删除操作,其中 m 为字符串的长度。这使得字典树在处理大量字符串时具有较高的效率。
4. **空间消耗**:尽管字典树在存储字符串数据时具有高效性,但其空间消耗较高。因为每个节点都需要存储指向子节点的指针,可能会导致内存空间的浪费。
5. **适用场景**:字典树适用于需要高效存储和检索大量字符串数据的场景,例如搜索引擎中的关键词提示、拼写检查、网络路由表等。
总的来说,字典树是一种用于高效存储和检索字符串数据的数据结构,通过利用字符串的前缀信息,能够快速地进行查找、插入和删除操作,具有较高的效率和适用性。
做题中,多用于字符串前缀的查找和字符串出现次数的查找。
当然还有很多功能需要结合题目来改变
当然,这里的cnt数组的作用可以改写改动,以适应不同题目要求的功能
int idx;
int cnt[N];//cnt[p]记录以节点p为结尾的字符串个数(是相同的字符串)
int ch[N][27];//ch[p][j]记录从p沿着j这条路到达的节点
void insert(string s) {
int p = 0;//根节点是0,从0开始
for (int i = 0; i < s.size(); i++) {
int j = s[i] - 'a';
if (!ch[p][j]) ch[p][j] = ++idx;
p = ch[p][j];
}
cnt[p]++;
}
int query(string s) {//查询s出现了几次
int p = 0;
for (int i = 0; i < s.size(); i++) {
int j = s[i] - 'a';
if (!ch[p][j]) return 0;
p = ch[p][j];
}
return cnt[p];
}
需要补充的是,字典树不只能存小写字母,也可以存大写字母,数字和二进制等等。
反正明白了原理就可以建树.
下面这道就是有关01字符串的。
P2922 [USACO08DEC] Secret Message G - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)
分析:相当于对两个字符串a,b判断a是不是b的前缀,b是不是a的前缀。
维护两个计数的数组(cnt,suf)即可
#include<iostream>
#include<map>
#include<set>
#include<cmath>
#include<vector>
#include<string>
#include<algorithm>
#include<queue>
#include<bitset>
#include<unordered_map>
using namespace std;
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0)
#define int long long
#define inf 0x3f3f3f3f
#define tans cout<<"答案:"
#define rep(i,a,b) for(int i = a;i<=b;i++)
const int N = 1e6 + 10;
//负数求模公式 (a % mod + mod) % mod
int gcd(int a, int b) {
return b > 0 ? gcd(b, a % b) : a;
}
int qmi(int a, int k, int p) {
int res = 1;
while (k) {
//后面的a其实是底数与其指数的运算结果了,是不断迭代的
//第一个a其实就是a的2的0次方
if (k & 1) res = (res * a) % p;
a = (a * a) % p;
//注意,a是一个不断变化的过程
//下一个a就等于上一个a的平方,
k >>= 1;
}
return res;
}
int mod = 1e9 + 7;
int C(int a, int b) {//这个就是根据定义求组合数
int ans = 1;
int j = a;
for (int i = 1; i <= b; i++, j--) {
ans = ans * j % mod;
ans = ans * qmi(i, mod - 2, mod) % mod;
}
return ans;
}
int idx;
int cnt[N];//cnt[p]记录以节点p为结尾的字符串个数(是相同的字符串)
int suf[N];//记录后缀
int ch[N][3];//ch[p][j]记录从p沿着j这条路到达的节点
void insert(string s) {
int p = 0;//根节点是0,从0开始
for (int i = 0; i < s.size(); i++) {
int j = s[i] - '0';
if (!ch[p][j]) ch[p][j] = ++idx;
p = ch[p][j];
suf[p]++;
}
cnt[p]++;
}
int query(string s) {//查询s出现了几次
int p = 0;
int ans = 0;
int f = 1;
for (int i = 0; i < s.size(); i++) {
int j = s[i] - '0';
if (!ch[p][j]) {
f = 0;
break;
}
p = ch[p][j];
if (i != s.size() - 1) ans += cnt[p];
}
if (f) {
ans += suf[p];
}
return ans;
}
void solve() {
int n, m; cin >> n >> m;
int span = '1' - 1;
rep(i, 1, n) {
int x; cin >> x;
string s;
rep(i, 1, x) {
int y; cin >> y;
s += y + span;
}
insert(s);
}
rep(i, 1, m) {
int x; cin >> x;
string s;
rep(i, 1, x) {
int y; cin >> y;
s += y + span;
}
//tans;
cout << query(s) << endl;
}
}
signed main() {
int t = 1;
IOS;
//cin >> t;
while (t--) solve();
return 0;
}
小技巧:如果一道题,涉及到字符串子串查找,前缀字符串匹配。而正好!题中说字符串只由小写字母组成(或者只由小写字母,大写字母,数字组成),那么就可以合理猜测一下这题是不是用字典树。