字典树模板

#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;
    }
};