我去,我连trie都不会写了,好尴尬,在丢下一个模板,来加强一下记忆
字典树应该原理都会吧,不进行赘述。
//这是map的一种手写版,但map是红黑树维护的,trie不能自动维护
struct _normal_trie{
int s[maxn][26],w[maxn];//节点以及其26个子节点,权值
int tot;
void init(){tot=0;}
void insert(char *a,int id){
int len=strlen(a),u=0;
for(int i=0;i<len;i++){
int t=a[i]-'a';//在树上进行寻找
u=s[u][t]?s[u][t]:s[u][t]=++tot;//如果没有此节点,就create一个,否则就使用那个节点
}
w[u]=id;//
}
int ask(char *a){
//查询同建立节点差不多
int len=strlen(a),u=0;
for(int i=0;i<len;i++){
int t=a[i]-'a';
if(s[u][t]==0) return 0;//如果不存在这个点,就没有必要继续遍历了
u=s[u][t];
}
return w[u];
}
}trie;