Trie字符串统计

最新推荐文章于 2025-12-08 11:07:14 发布

原创最新推荐文章于 2025-12-08 11:07:14 发布 · 105 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#c语言 #数据结构

数据结构专栏收录该内容

12 篇文章

订阅专栏

本文介绍了一种使用Trie树实现字符串插入与查询的方法。通过定义son数组记录子节点ID，cnt数组记录以特定节点结尾的字符串数量，实现了高效的数据结构。详细解释了insert与query函数的工作原理及时间复杂度的优势。

#include <iostream>
using namespace std;

const int N = 1e5 + 10;
int son[N][26]; // 其中存放的是：子节点对应的idx。其中son数组的第一维是：父节点对应的idx，第第二维计数是：其直接子节点('a' - '0')的值为二维下标。
int cnt [N];    // 以“abc”字符串为例，最后一个字符---‘c’对应的idx作为cnt数组的下标。数组的值是该idx对应的个数。
int idx;        // 将该字符串分配的一个树结构中，以下标来记录每一个字符的位置。方便之后的插入和查找。
char str[N];

void insert(char *str)
{
    int p = 0;
    for (int i = 0; str[i]; i++)
    {
        int u = str[i] - '0';
        if (!son[p][u]) son[p][u] = ++idx;
        p = son[p][u];
    }
    // 此时的p就是str中最后一个字符对应的trie树的位置idx。
    cnt[p]++;
}

int query(char *str)
{
    int p = 0;
    for (int i = 0; str[i]; i++)
    {
        int u = str[i] - '0';
        if (!son[p][u]) return 0;
        p = son[p][u];
    }
    return cnt[p];
}

int main()
{
    int n;
    scanf("%d", &n);
    char op[2];
    while (n--) 
    {
        scanf("%s%s", op, str);
        if (op[0] == 'I') insert(str);
        else printf("%d\n", query(str));
    }

    return 0;
}