trie树(字典树)--模板

我去,我连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;
发布了51 篇原创文章 · 获赞 6 · 访问量 2万+

猜你喜欢

转载自blog.csdn.net/qq_35776579/article/details/53413739