字典树模板

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <vector>
#include <algorithm>
#include <string>
#include <iostream>
using namespace std;
const int MAXN=500000+10;
struct trie{
int ch[MAXN<<3][30];
int val[MAXN<<3];
int sz;
void init(){
sz=1;
memset(ch[0],0,sizeof ch[0]);
memset(val,0,sizeof val);
}
int idx(char c){
return c-'a';
}
void insert(const char *s){
int u=0,n=(int)strlen(s);
for(int i=0;i<n;i++){
int c=idx(s[i]);
if(!ch[u][c]){
memset(ch[sz],0,sizeof ch[sz]);
val[sz]=0;
ch[u][c]=sz++;
}
u=ch[u][c];
val[u]++;
}
}
int query(const char *s,int rt,int n,int cur){
int res=0;
if(cur==n){
return val[rt];
}
if(s[cur]=='$'){
for(int i=0;i<26;i++){
if(ch[rt][i]){
res+=query(s, ch[rt][i], n, cur+1);
}
}
}else{
int c=idx(s[cur]);
if(ch[rt][c]){
return query(s,ch[rt][c],n,cur+1);
}
return 0;
}
return res;
}
};