字典树

//----------用数组实现a -> z的字典树----------
//注：若next报错则用nxt等其他变量名
const int maxn = 100010;
const int charsize = 26;    //字符集大小

//记录此节点的子节点编号（默认全是小写字母）
int next[maxn + 1][charsize];

//表示此编号节点是否为终止节点
bool isEnd[maxn + 1];

int root = 0;
int cnt = 0;    //表示当前的节点编号数

//----------字典树的插入----------

//s为等待插入的字符串，1base，len为字符串长度
void insert(char s[], int len){
    int now = 0;    //当前在哪个节点上
    for (int i = 1;i <= len;i++){   //遍历，没什么好说的
        int x = s[i] - 'a';     //转成数字
        if (!next[now][x]){     //如果当前节点没有子节点x，则创建之，并且给予编号
            next[now][x] = ++cnt;  
        }
        now = next[now][x];     //无论创建了新节点与否，都更新当前节点的位置
    }
    isEnd[now] = true;          //最后的最后，把该字符串的最后一个字符的isEnd标记为true
    return;
}

//----------字典树的查找----------

//实际上查找和插入的本质是相似的
//s为等待插入的字符串，1base，len为字符串长度
bool search(char s[], int len){
    int now = 0;    //和上面一样，记录当前在看的节点
    for (int i = 1; i <= len;i++){  //遍历之
        int x = s[i] - 'a';     //转成数字
        if (!next[now][x]){     //如果当前节点没有子节点x，则直接返回false
            return false;
        }
        now = next[now][x];     //如果没有return则更新当前节点位置，继续循环
    }
    return isEnd[now];          //最后进行判断最后一个字符是否是结束位置，返回即可
}

//----------字典树的删除----------

//其实Trie树需要删除数据的情况比较罕见
//这个时候我们需要记录每个树节点的"子树大小"，这里的子树大小不同于一般的概念（一般的就是说子树有多少个节点）
//这里的子树大小指的是该节点的子树上有多少个真实存在的字符串（真实存在的意思就是必须结尾字符的isEnd是true）

//对于要删除的字符串，先用查找字符串的方式找到末尾，删除标记（isEnd）后回溯整条路径
//值得一提的是，如果路径上节点的子树大小为0，就可以直接删除这个节点




//----------用Struct存储字典树----------

const int maxn = 100010;
const int charsize = 26;

//-----简单写法-----

struct TreeNode{
    int next[charsize];
    bool isEnd;
} tree[maxn + 1];

//-----当字符集较大时-----

struct TreeNode{
    unordered_map <char, int> next_map;
    bool isEnd;
} tree[maxn + 1];

//-----指针-----

struct TreeNode{
    TreeNode *next[charsize];
    bool isEnd;
} *root;