Trie模版

Posted 2020-10-15 white_hat_hacker

struct Trie{
    Trie* nxt[26];
    int v;
    Trie(){
        for(int i=0;i<26;i++){
            nxt[i]=NULL;
        }
        v=-1;
    }
    void insert(char s[],int vl){
        Trie *p=this;
        int len=strlen(s);
        for(int i=0;i<len;i++){
            int t=s[i]-97;
            if(p->nxt[t]){
                p=p->nxt[t];
            }
            else{
                p->nxt[t]=new Trie;
                p=p->nxt[t];
            }
        }    
        p->v=vl;
    }
    int find(char s[]){
        Trie *p=this;
        int len=strlen(s);
        for(int i=0;i<len;i++){
            int t=s[i]-97;
            if(p->nxt[t]){
                p=p->nxt[t];
            }
            else{
                return -1;
            }
        }
        return p->v;
    }
}T;