字典树模板

原理详见：https://www.cnblogs.com/TheRoadToTheGold/p/6290732.html

模板一：

输入字符串，查询是不是单词或者前缀

查询单词的时候，在插入时在最后一个节点的地方标记为单词，查询时返回最后节点是不是单词标记

const int maxn = 1e6 + 10;
int tree[maxn][26];
//字典树tree[u][v]表示编号为u的节点的下一个字母为v连接的节点的编号
int idx(char c){ return c - 'a'; }//可以写成宏定义
int tot = 1;//根节点编号为1
bool is_word[maxn];//单词结束标记
void Insert(char s[], int u)//u表示根节点
//插入字符串s
{
    for(int i = 0; s[i]; i++)
    {
        int c = idx(s[i]);
        if(!tree[u][c])
            tree[u][c] = ++tot;
        u = tree[u][c];
    }
    //is_word [u] = true; //查询单词的时候需要标记最后一个节点的地方是单词
}

bool Find(char s[], int u)
//查询s是否是前缀
{
    for(int i = 0; s[i]; i++)
    {
        int c = idx(s[i]);
        if(!tree[u][c])
            return false;
        u = tree[u][c];
    }
    return true;
    //return is_word[u]; //查询s是否是单词的时候，需要返回当前是不是单词结束标志
}

统计前缀出现的次数：

由于字典树中字母是边，节点是编号，统计前缀出现次数时，开一个sum数组表示该编号为终点的前缀出现次数，在每次加入一个字母的时候，总是在这条字母表示的边的后面那个节点加一。

比如下图，红色表示每个节点为终点的前缀出现的次数，每个字母表示边。

这里在每条边的后面那个节点上表示该前缀出现的次数。

如果在每条边的前面那个节点上表示该前缀出现的次数，会导致错误，因为根节点就不知道到底表示成什么字母出现的次数

查询c为前缀的数目的时候返回编号1的sum值

查询ca为前缀的数目的时候返回编号2的sum值

const int maxn = 1e6 + 10;
int tree[maxn][26];
//字典树tree[u][v]表示编号为u的节点的下一个字母为v连接的节点的编号
int idx(char c){ return c - 'a'; }//可以写成宏定义
int tot = 1;//根节点编号为1
int sum[maxn];//标记以该节点结束的前缀出现次数
void Insert(char s[], int u)//u表示根节点
//插入字符串s
{
    for(int i = 0; s[i]; i++)
    {
        int c = idx(s[i]);
        if(!tree[u][c])
            tree[u][c] = ++tot;
        sum[tree[u][c]]++; //前缀后面的那个节点数目加一
        u = tree[u][c];
    }
}

int Find_sum(char s[], int u)
{
    for(int i = 0; s[i]; i++)
    {
        int c = idx(s[i]);
        if(!tree[u][c])return 0;
        u = tree[u][c];
    }
    return sum[u];//返回最后一个字母表示的边连接的后面那个节点，所记录的sum值
}

 删除包含某前缀的所有单词

void Delete(char s[], int u)//删除所有包含该前缀的单词
{
    int cnt = Find_sum(s, u);
    if(cnt <= 0)return;

    for(int i = 0; s[i]; i++)//前缀路径上全部删去cnt个单词
    {
        int c = idx(s[i]);
        if(!tree[u][c])return;
        sum[u] -= cnt;
        u = tree[u][c];
    }
    sum[u] = 0;//最末尾直接赋值成0
    for(int i = 0; i < 26; i++)//最末尾的每个子节点全部清空
        tree[u][i] = 0;
}