Trie 字典树

Trie

Trie（字典树）是一种用于实现字符串快速检索的多叉树结构。Tire的每个节点都有若干个字符指针，若再插入和检索字符串时扫描到一个字符，就沿字符走向指向的节点。

 

插入

当需要插入一个字符串时，我们从根走起，依次扫描每一个字符：

1. 若字符指向一个已经存在的节点，则顺着节点继续走下去
2. 若字符指向空，则新建一个节点，继续走下去。

当字符串中的字符都检索完毕时，记录下其终点结点。

检索

当需要检索一个字符串是否存在时，我们从根走起，一次扫描每一个字符：

1. 若字符指向空，则说明不存在，结束。
2. 若字符指向一个结点，则继续走下去。

 

当字符串中所有字符被检索完毕时，若当前结点是一个被标记为终点的节点，说明字符串存在，否则说明其不存在。

int trie[SIZE][26], tot = 1;
void insert(char* str){
    int len = strlen(str), p = 1;
    for (int k = 0; k < len; k++){
        int ch = str[k] - 'a';
        if (trie[p][ch] == 0) trie[p][ch] = ++tot;
        p = trie[p][ch];
    }
    end[p] = true;
}
bool search(char* str){
    int len = strlen(str), p = 1;
    for (int k = 0; k <= len; k++){
        p = trie[p][str[k] - 'a'];
        if (p == 0) return false;
    }
    return end[p];
}