Trie树 & 01Trie

指针版

#define MAXNUM 26
//定义字典树结构体
typedef struct Trie
{
    bool flag;//从根到此是否为一个单词
    Trie *next[MAXNUM];
}Trie;
//声明一个根
Trie *root;
//初始化该根
void init()
{
    root = (Trie *)malloc(sizeof(Trie));
    root->flag=false;
    for(int i=;i<MAXNUM;i++)
    root->next[i]=NULL;
}
//对该字典树的插入单词操作
void insert(char *word)
{
    Trie *tem = root;
    while(*word!='\0')
    {
        if(tem->next[*word-'a']==NULL)
        {
            Trie *cur = (Trie *)malloc(sizeof(Trie));
            for(int i=;i<MAXNUM;i++)
            cur->next[i]=NULL;
            cur->flag=false;
            tem->next[*word-'a']=cur;
        }
        tem = tem->next[*word-'a'];
        word++;
    }
    tem->flag=true;
}
//查询一个单词的操作
bool search(char *word)
{
    Trie *tem = root;
    for(int i=;word[i]!='\0';i++)
    {
        if(tem==NULL||tem->next[word[i]-'a']==NULL)
        return false;
        tem=tem->next[word[i]-'a'];
    }
    return tem->flag;
}
//释放字典树内存操作，由于本题测试数据后程序自动跳出，所以这里没写释放内存函数
void del(Trie *cur)
{
    for(int i=;i<MAXNUM;i++)
    {
        if(cur->next[i]!=NULL)
        del(cur->next[i]);
    }
    free(cur);
}

01Trie

 struct Trie
 {
     int root, tot, next[*][], cnt[*], end[*];

     inline int Newnode()
     {
         memset(next[tot], -, sizeof(next[tot]));
         cnt[tot] = ;
         end[tot] = ;
         return tot ++;
     }

     inline void Init()
     {
         tot = ;
         root = Newnode();
     }

     inline void Insert(int x)
     {
         int p = root;
         cnt[p] ++;
         for(int i = ; i >= ; i --)
         {
             int idx =  (x >> i) & ;
             if(next[p][idx] == -)
                 next[p][idx] = Newnode();
             p = next[p][idx];
             cnt[p] ++;
         }
         end[p] = x;
     }

     inline void Del(int x)
     {
         int p = root;
         cnt[p] --;
         for(int i = ; i >= ; i --)
         {
             int idx =  (x >> i) & ;
             p = next[p][idx];
             cnt[p] --;
         }
     }

     inline int Search(int x)  //求x的异或最大值，看情况修改
     {
         int p = root;
         for(int i = ; i >= ; i --)
         {
             int k = (x >> i) & ;
             if(next[p][k^] != - && cnt[next[p][k^]])
                 p = next[p][k^];
             else
                 p = next[p][k];
         }
         return x ^ end[p];
     }
 }tr;