tire树代码示例和例题

tire树

【模板】字典树

题目描述

给定 $n$ 个模式串 $s_1,s_2,\dots ,s_n$ 和 $q$ 次询问，每次询问给定一个文本串 $t_i$，请回答 $s_1\sim s_n$ 中有多少个字符串 $s_j$ 满足 $t_i$ 是 $s_j$ 的前缀。

一个字符串 $t$ 是 $s$ 的前缀当且仅当从 $s$ 的末尾删去若干个（可以为 0 个）连续的字符后与 $t$ 相同。

输入的字符串大小敏感。例如，字符串 Fusu 和字符串 fusu 不同。

输入格式

本题单测试点内有多组测试数据。

输入的第一行是一个整数，表示数据组数 $T$。

对于每组数据，格式如下：
第一行是两个整数，分别表示模式串的个数 $n$ 和询问的个数 $q$。
接下来 $n$ 行，每行一个字符串，表示一个模式串。
接下来 $q$ 行，每行一个字符串，表示一次询问。

输出格式

按照输入的顺序依次输出各测试数据的答案。
对于每次询问，输出一行一个整数表示答案。

样例 #1

样例输入 #1

3
3 3
fusufusu
fusu
anguei
fusu
anguei
kkksc
5 2
fusu
Fusu
AFakeFusu
afakefusu
fusuisnotfake
Fusu
fusu
1 1
998244353
9

样例输出 #1

2
1
0
1
2
1

提示

数据规模与约定

对于全部的测试点，保证 $1\le T,n,q\le 10^5$，且输入字符串的总长度不超过 $3\times 10^6$。输入的字符串只含大小写字母和数字，且不含空串。

说明

std 的 IO 使用的是关闭同步后的 cin/cout，本题不卡常。

#include <iostream>
#include <string.h>
using namespace std;
const int N = 1e5 + 10;
int son[N][65], idx, cnt[N]; // 实现trie树的相关数据结构
int getnum(char x);
void insert(char str[])
{
    int p = 0; // 表示根节点
    int len = strlen(str);
    for (int i = 0; i < len; i++)
    {
        int u = getnum(str[i]);
        if (!son[p][u])
            son[p][u] = ++idx;
        p = son[p][u];
        cnt[p]++;
    }
    // cnt[p]++;
}
int search(char str[])
{
    int p = 0; // 表示根节点
    int len = strlen(str);
    for (int i = 0; i < len; i++)
    {
        int u = getnum(str[i]);
        if (!son[p][u])
            return 0;
        p = son[p][u];
    }
    return cnt[p];
}

int getnum(char x)
{
    if (x >= 'A' && x <= 'Z')
        return x - 'A';
    else if (x >= 'a' && x <= 'z')
        return x - 'a' + 26;
    else
        return x - '0' + 52;
}
int main()
{
    int n;
    cin >> n;
    char s[N];
    for (int i = 0; i < n; i++)
    {
        for (int i = 0; i <= idx; i++)
            for (int j = 0; j <= 122; j++)
                son[i][j] = 0;
        for (int i = 0; i <= idx; i++)
            cnt[i] = 0;
        idx = 0;

        int x, y;
        cin >> x >> y;
        for (int k = 0; k < x; k++)
        {
            cin >> s;
            insert(s);
        }
        for (int k = 0; k < y; k++)
        {
            cin >> s;
            cout << search(s) << endl;
        }
    }
    return 0;
}