算法竞赛进阶指南---0x16 （Trie）前缀统计

最新推荐文章于 2025-08-06 17:01:48 发布

原创最新推荐文章于 2025-08-06 17:01:48 发布 · 189 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #数据结构 #字典树

Trie字典树专栏收录该内容

6 篇文章

订阅专栏

本文介绍如何使用字典树（Trie）高效存储和查询字符串集合。通过构建字典树并添加特殊标记，文章详细解释了如何进行字符串插入和前缀查询的过程，并提供了完整的C++代码实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题面

在这里插入图片描述

题解

Trie(字典树) ：高效地存储和查找字符串集合的数据结构( O(n) )
我们可以先将N个字符串用字典树的方式存储起来，并且把每个字符串末尾的位置添加标记（+1的操作，代表以这个节点结尾的是一个字符串）
然后对于M次询问，可以查询字典树，从根节点向下遍历的过程中，遇到有节点标记的，就说明是改字符串的前缀
易错点：N个字符串可能有重复，所以只能统计以该节点结尾的字符串的个数，不能统计是否以该节点结尾
建树模拟过程如下图

代码

#include<bits/stdc++.h>

using namespace std;
const int N = 1e6 + 10;

int n, m;
int son[N][26];  //储存树中每个节点的子节点
int cnt[N];     //存储以每个节点结尾的单词数量
int idx;   //下标是0的点，既是根节点，又是空节点

//插入一个字符串
void insert(char str[]) {
    int res = 0;
    int p = 0;
    for (int i = 0; str[i]; i++) {
        int u = str[i] - 'a';
        if (!son[p][u]) son[p][u] = ++idx;
        p = son[p][u];
    }
    cnt[p]++;
}

//查询一个字符串
int query(char str[]) {
    int res = 0;
    int p = 0;
    for (int i = 0; str[i]; i++) {
        int u = str[i] - 'a';
        if (!son[p][u]) return res;
        p = son[p][u];
        if (cnt[p] > 0) res+=cnt[p];
    }

    return res;
}


int main() {

    cin >> n >> m;

    for (int i = 1; i <= n; i++) {
        char s[N];
        cin >> s;
        insert(s);
    }

    for (int i = 1; i <= m; i++) {
        char s[N];
        cin >> s;
        cout << query(s) << endl;
    }

    return 0;
}