视频链接：F06 字典树(Trie)

0）概述

• 是快速插入和查询字符串的多叉树结构，根节点编号为0，其余节点标识路径，还可以标记单词插入的次数，边表示字符。

1）数据结构

const int N=1e5+5;
char s[N]; // 每次输入的字符串,N是每个单词的最大长度
int ch[N][26]; // ch[p][j]:从节点p沿着j这条边走到的子节点,边为26个小写字母映射值为0~25
int cnt[N]; // cnt[p]:以节点p结尾的单词的插入次数
int idx; // 遍历因子

2）插入操作

• insert函数，插入单个单词并建立字典树

// s:单词(字符串)
void insert(char *s) {int p=0; // 根节点编号为0// 枚举字符串每个字符for(int i=0;s[i];i++) {int j=s[i]-'a'; // a~z映射到0~25// 如果这个字符不是儿子节点,创建儿子,p指针再走到儿子if(!ch[p][j]) ch[p][j]=++idx; // 节点编号+1// 如果这个字符是儿子节点,p指针走到儿子节点p=ch[p][j];}cnt[p]++; // 以节点p结尾的单词插入的次数+1 
}

3）查询操作

• query函数，得到一个单词被插入的次数

// 查询某个单词出现的
int query(char *s) {int p=0; // 从根节点开始查// 扫描字符串for(int i=0;s[i];i++) {int j=s[i]-'a'; // 转换为映射值if(!ch[p][j]) return 0; // 如果找不到返回0// 有字母s[i],则走下来p=ch[p][j];}// 如果能走到词尾,则返回插入次数return cnt[p];
}

4）完整代码

1. 字符数组

#include<bits/stdc++.h>
#define x first
#define y secondusing namespace std;typedef long long ll;
typedef pair<int,int> PII;// 解题思路: 节点表示数字,边表示字符const int N=1e5+5;
char s[N]; // 每次输入的字符串,N是每个单词的最大长度
int ch[N][26]; // ch[p][j]:从节点p沿着j这条边走到的子节点,边为26个小写字母映射值为0~25
int cnt[N]; // cnt[p]:以节点p结尾的单词的插入次数
int idx; // 遍历因子// s:单词(字符串)
void insert(char *s) {int p=0; // 从根节点开始插// 枚举字符串每个字符for(int i=0;s[i];i++) {int j=s[i]-'a'; // a~z映射到0~25// 如果这个字符不是儿子节点,创建儿子,p指针再走到儿子if(!ch[p][j]) ch[p][j]=++idx; // 节点编号+1// 如果这个字符是儿子节点,p指针走到儿子节点p=ch[p][j];}cnt[p]++; // 以节点p结尾的单词插入的次数+1 
}// 查询某个单词出现的
int query(char *s) {int p=0; // 从根节点开始查// 扫描字符串for(int i=0;s[i];i++) {int j=s[i]-'a'; // 转换为映射值if(!ch[p][j]) return 0; // 如果找不到返回0// 有字母s[i],则走下来p=ch[p][j];}// 如果能走到词尾,则返回插入次数return cnt[p];
}int main() {int n;cin>>n;while(n--) {char op;scanf("%s%s",&op,s);if(op=='I') insert(s);else cout<<query(s)<<'\n';}return 0;
}

2. 字符串

#include<bits/stdc++.h>
#define x first
#define y secondusing namespace std;typedef long long ll;
typedef pair<int,int> PII;// 解题思路: 节点表示数字,边表示字符// 文件总长度不超过32K,所以总字符不超过:32K=32*1024
const int N=32*1024+10; 
char s[N]; // 每次输入的字符串,N是每个单词的最大长度
int ch[N][26]; // ch[p][j]:从节点p沿着j这条边走到的子节点,边为26个小写字母映射值为0~25
int cnt[N]; // cnt[p]:以节点p结尾的单词的插入次数
int idx; // 遍历因子// s:单词(字符串)
void insert(string s) {int p=0; // 从根节点开始插// 枚举字符串每个字符for(int i=0;i<s.length();i++) {int j=s[i]-'A'; // a~z映射到0~25// 如果这个字符不是儿子节点,创建儿子,p指针再走到儿子if(!ch[p][j]) ch[p][j]=++idx; // 节点编号+1// 如果这个字符是儿子节点,p指针走到儿子节点p=ch[p][j];}cnt[p]++; // 以节点p结尾的单词插入的次数+1 
}// 查询某个单词出现的
int query(string s) {int p=0; // 从根节点开始查// 扫描字符串for(int i=0;i<s.length();i++) {int j=s[i]-'a'; // 转换为映射值if(!ch[p][j]) return 0; // 如果找不到返回0// 有字母s[i],则走下来p=ch[p][j];}// 如果能走到词尾,则返回插入次数return cnt[p];
}int main() {
//	while(scanf("%s",s)) {
//		insert(s);
//	}ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);string s;while(cin>>s) {insert(s);}cout<<idx+1; // 加上根节点return 0;
}