CH 1601前缀统计

版权声明：本文为博主原创文章，记录蒟蒻的成长之路，欢迎吐槽~ https://blog.csdn.net/PegasiTIO/article/details/89420051

描述

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

输入格式

第一行两个整数N，M。接下来N行每行一个字符串Si。接下来M行每行一个字符串表示询问。

输出格式

对于每个询问，输出一个整数表示答案

样例输入

3 2
ab
bc
abc
abc
efg

样例输出

2
0

---

字典树

记录每个结点是多少个字符串的末尾结点，查找时累加每个点的数量就可以了

#include <cstdio>
#include <cstring>
using namespace std;

const int maxn = 1000005;
int tot = 1, trie[maxn][26], cnt[maxn];
char s[maxn];

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];
    }
    ++cnt[p]; //出现数量加1
}

int check(char *str)
{
    int len = strlen(str), p = 1, ans = 0;
    for (int k = 0; k < len; ++k)
    {
        p = trie[p][str[k] - 'a']; //下一个节点
        if (p == 0)
            return ans;
        ans += cnt[p]; //加上这个点的字符串数量
    }
    return ans;
}

int main()
{
    int n, m;
    scanf("%d%d", &n, &m);
    for (int i = 0; i < n; ++i)
    {
        scanf("%s", s);
        insert(s);
    }

    for (int i = 0; i < m; ++i)
    {
        scanf("%s", s);
        printf("%d\n", check(s));
    }
    return 0;
}