Trie 字符串
Trie字符串:快速存储字符串集合
那么它是怎么样的一种结构呢??
特点:字母的个数不会很多,why?
来看代码的实现吧
题目
AcWing835 Trie字符串
维护一个字符串集合,支持两种操作:
“I x”向集合中插入一个字符串x;
“Q x”询问一个字符串在集合中出现了多少次。
共有N个操作,输入的字符串总长度不超过 105,字符串仅包含小写英文字母。
模板
static int[][] son = new int[N][26];//只存储小写字母
static int[] cnt = new int[N]; //存取下标
static int idx; //idx索引,唯一
private static void insert(String s){
//存储一个字符串
int p = 0;
for(int i=0;i<s.length();i++){
int u = s.charAt(i) - 'a';//存个差值就ok了
if(son[p][u]==0){
//如果子节点没有,就新增
son[p][u] = ++idx;
}
p = son[p][u];//下一层
}
//存储完毕后,记录下标
cnt[p]++;
}
private static int query(String s){
//查询一个字符串
int p = 0;
for(int i=0;i<s.length();i++){
int u = s.charAt(i)-'a';
if(son[p][u]==0){
return 0;
}
p = son[p][u];
}
return cnt[p];
}