【暑假】[实用数据结构]前缀树 Trie

前缀树Trie

 

Trie可理解为一个能够快速插入与查询的集合，无论是插入还是查询所需时间都为O(m)

 

模板如下：

 

const int maxnode = 1000+10;
const int sigma_size = 26; 

struct Trie{
int ch[maxnode][sigma_size];
int val[maxnode];
int sz;

  void clear(){ sz=1; memset(ch[0],0,sizeof(ch[0])); }
  int ID(char c){ return c='a'; };
  
  void insert(char* s,int v){ //向Trie中插入 
      int u=0 , n=strlen(s);
      for(int i=0;i<n;i++) { 
          int c=ID(s[i]);
          if(!ch[u][c]){
              memset(ch[sz],0,sizeof(ch[sz])); //初始化子结点组 
              val[sz]=0;        //中间结点val=0 
              ch[u][c]= sz++;  //建立子结点 
        }
        u=ch[u][c];          //往下走 
      }
      val[u]=v;
  }
  
  int find(char *s){     //查找s是否存在 
      int n=strlen(s);
      int u=0;
      for(int i=0;i<n;i++){
          int c=ID(s[i]);
          if(!ch[u][c]) return false;
          u=ch[u][c];
      }
      return val[u];  //是否为尾 
  }
};