《算法竞赛进阶指南》1.6Trie

142. 前缀统计

给定N个字符串S1,S2…SN，接下来进行M次询问，每次询问给定一个字符串T，求S1～SN中有多少个字符串是T的前缀。
输入字符串的总长度不超过106，仅包含小写字母。

输入格式
第一行输入两个整数N，M。
接下来N行每行输入一个字符串Si。
接下来M行每行一个字符串T用以询问。

输出格式
对于每个询问，输出一个整数表示答案。
每个答案占一行。

输入样例：
3 2
ab
bc
abc
abc
efg

输出样例：
2
0

#include <iostream>

using namespace std;

const int N = 1000010, M = 500000; //trie 的长度

int n, m;
int son[M][26], cnt[N], idx; //cnt记录该节点是多少个字符串的末尾节点 idx 虚拟内存的指针(当前用到的数组的地址，当前节点的下标)
char str[N];

int query() //查询 字符串前缀个数
{
   
    int p = 0, res = 0; //res 表示当前路径所有单词的总和
    for(int i = 0; str[i]; i++)
    {
   
        int &s = son[p][str[i] - 'a'];
        if(!s) break; //当前节点为空 没有下面的路径 break
        p = s;
        res += cnt[p]; //加上以当前节点为结尾的个数
    }
    return res;
}

void insert() //插入
{
   
    int p = 0; //trie根节点  0号节点
    for(int i = 0; str[i]; i++)
    {
   
        int &s = son[p][str[i] - 'a']; //当前儿子
        if(!s) s = ++ idx; //s 不存在 分配一个新的值
        p = s; 
    }
    cnt[p] ++; //有一个单词以p节点为结尾 加1
}

int main()
{