【ACWing】142. 前缀统计

题目地址：

https://www.acwing.com/problem/content/144/

给定$N$个字符串$S_1,S_2,\dots ,S_N$，接下来进行$M$次询问，每次询问给定一个字符串$T$，求$S_1\sim S_N$中有多少个字符串是$T$的前缀。输入字符串的总长度不超过$10^6$，仅包含小写字母。

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

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

直接用Trie。代码如下：

#include <iostream>
using namespace std;

const int N = 1e6 + 10;
int n, m;
int son[N][26], cnt[N];
char s[N];
int idx;

void insert() {
    
    
    int p = 0;
    for (int i = 0; s[i]; i++) {
    
    
        int u = s[i] - 'a';
        if (!son[p][u]) son[p][u] = ++idx;
        p = son[p][u];
    }

    cnt[p]++;
}

int query() {
    
    
    int p = 0, res = 0;
    for (int i = 0; s[i]; i++) {
    
    
        int u = s[i] - 'a';
        if (son[p][u]) p = son[p][u];
        else break;

        res += cnt[p];
    }

    return res;
}

int main() {
    
    
    scanf("%d%d", &n, &m);
    while (n--) {
    
    
        cin >> s;
        insert();
    }
    
    while (m--) {
    
    
        cin >> s;
        printf("%d\n", query());
    }

    return 0;
}

每次询问时间复杂度$O(l_s)$，空间$O(N)$。