Trie树模版

Trie 树 中文名叫字典树，可以用来存放n个单词，并且找出某个前缀的数量，或者找出某个单词的数量。

其实也有其他的应用，比如统计有多少个不同的字符串等等。

字典树分为一般分为两个部分，一个是创建字典树，还一个是find函数，find函数的写法随着题目要求可以灵活多变的！

首先 要先定义出数据结构 。

 struct Trie{
Trie *next[26]; //指向子节点
int v; //可以很多个变量，比如可以统计这个前缀的数量，或者标记是不是结尾等等
};
Trie *root;

 

初始化函数：

void init()
{
root = (Trie*)malloc(sizeof(Trie));
for(int i = 0;i < 26;i++)
root->next[i] = NULL;
}

 

创建字典树：

void creat_Trie(char *str)  //创建字典树 
{ 
    int len = strlen(str);  //字符串的长度 
    Trie *p = root, *q;// p 指向字典树，定义一个指针q 
    for(int i=0; i<len; ++i)//遍历整个字符串 
    {
        int id = str[i]-'0';  //得到每个字符的值 
        if(p->next[id] == NULL)//如果该值在字典树中是第一次出现 
        {
            q = (Trie *)malloc(sizeof(Trie));//开辟一个新节点 
            q->v = 1;    //初始v==1,遇到一个新节点，则把经过这个节点的字符串数量初始化为1 
            for(int j=0; j<MAX; ++j)//让其的每种可能的下一个节点都为空 
                q->next[j] = NULL;
            p->next[id] = q;//让p指向过去  
            p = p->next[id];//p指向开节点的后一节点 
        }
        else
        {
            p->next[id]->v++;//如果存在，则让v++ ， 意思是通过这个节点的字符串数量又多了一个 
            p = p->next[id];//下一个节点。。 
        }
    }
     // p->t = -1;   //若为结尾，则将v改成-1表示(视情况而定)
}

 

 

 find函数：

 

int findTrie(char *str)
{
int len = strlen(str);   //查找这个串是否在字典树中 
Trie *p = root;//让p 指向字典树的一个节点。 
for(int i=0; i<len; ++i)
{
int id = str[i]-'0';  //根据需要选择是减去'0'还是'a'，或者是'A'
p = p->next[id]; //注意，如果这里指向下一届点了，下面判断就只能是p == NULL  如果要是p->next[id] == NULL 就应该在循环末尾进行。
if(p == NULL)   //若为空集，表示不存以此为前缀的串
return 0;
if(p->v == -1)   //字符集中已有串是此串的前缀
return -1;
}
return -1;   //此串是字符集中某串的前缀
} 

 

注意指针的 难调试性，写的时候千万不能写错了。。。

 

llj:

/*
ch[i][j]表示： i结点的j编号，如果该值为0，则没有j编号这个结点，比如 i = 1（即该字符为b），j=2（字符对应c），则 ch[i][j]:在b字母下面没有c，即还没有含有bc的字符串。 
*/
struct Trie
{
    int ch[maxnode][sigma_size];
    int val[maxnode];
    int sz;
    Trie() : sz(1) { memset(ch[0], 0, sizeof(ch[0])); }
    void reset() { memset(ch, 0, sizeof(ch)); memset(val, 0, sizeof(val)); sz = 1; }
    int idx(char c) { return c - 'a'; }

    void insert(char * s, int v)
    {
        int u = 0;
        for(int i = 0; s[i]; ++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] = v;
    }

    int query(char * s, int a) 
    {
        int u = 0, res = 0;
        for(int i = a; s[i]; ++i)
        {
            int c = idx(s[i]);
            if(!ch[u][c]) return res;
            u = ch[u][c];
            if(val[u])  // 查询到单词节点
            {
                ； 
            }
        }
        return res;
    }
}T;