字典树的应用背景:
看以下几个题:
1、给出n个单词和m个询问,每次询问一个单词,回答这个单词是否在单词表中出现过
答:简单!map,短小精悍。
好。下一个
每次询问一个前缀,回答询问是多少个单词的前缀。2、给出n个单词和m个询问,
答:map,把每个单词拆开。
judge:n<=200000,TLE!
这就需要一种高级数据结构-Trie树(字典树)
截图的是--------------------------->Bilibili:董晓算法的内容
代码里面a是存储Trie树节点用的,a[p][u],里面u是表示英文字符,
字符
‘a’->0,
‘b’->1,
‘c’->2
p是代表第p个节点,a[p][u]值为k,那么k仅代表了编号
cnt[p]是代表了以p节点结尾这个单词出现次数!
这个trie树的知识点比较固定,似乎没有什么东西,死记硬背就好了
#include<iostream>
#include<string>
using namespace std;
#define MAX 100086
int N,a[MAX][27],idx,cnt[MAX];
string x,op;
int query(string& ele){
int p=0;
for(int i=0;ele[i];++i){
int u=ele[i]-'a';
if(!a[p][u]){
return 0;
}
p=a[p][u];
}
return cnt[p];
}
void insert(string& ele){
int p=0;
for(int i=0;ele[i]!='\0';++i){
int u=ele[i]-'a';
if(!a[p][u]){
a[p][u]=++idx;
}
p=a[p][u];
}
cnt[p]++;
}
int main(){
cin>>N;
while(N--){
cin>>op>>x;
if(op=="Q"){
cout<< query(x) <<endl;
}else if(op=="I"){
insert(x);
}
}
return 0;
}