【HDU 1251 统计难题】字典树(Trie树) 模板题

最新推荐文章于 2023-12-15 10:02:56 发布

原创最新推荐文章于 2023-12-15 10:02:56 发布 · 293 阅读

0 ·

CC 4.0 BY-SA版权

HDU 同时被 2 个专栏收录

33 篇文章

订阅专栏

字典树

17 篇文章

订阅专栏

本文介绍了一个基于Trie树的数据结构解决字符串前缀匹配问题的经典案例——HDU1251题目。通过构建Trie树，并记录每个节点的访问次数，实现了高效查询特定字符串作为前缀的单词数量。

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

HDU1251
题意就是统计出以某个字符串为前缀的单词数量，首先构建出trie树并记录每个节点的访问次数，然后在上面查询就好了，模板题。
HDU1251代码

#include<stdio.h>
#include<iostream>
#include<string.h>
using namespace std;
const int maxn =2e6+5;
int tree[maxn][30];
int sum[maxn];
int tot;
void insert_(char *str)
{
   int  len=strlen(str);
   int root=0;
   for(int i=0;i<len;i++)
   {
       int id=str[i]-'a';
       if(!tree[root][id]) tree[root][id]=++tot;
       sum[tree[root][id]]++;//记录节点访问次数
       root=tree[root][id];
   }
   //root在此对应某个单词，一一对应
}
int find_(char *str)
{
    int len=strlen(str);
    int root=0;
    for(int i=0;i<len;i++)
    {
        int id=str[i]-'a';
        if(!tree[root][id]) return 0;
        root=tree[root][id];
    }
    return sum[root];//返回当前字符串结尾节点的访问次数，也就是作为前缀的出现次数
}
char ss[maxn];
int main()
{
    tot=0;
    while(gets(ss))
    {
        if(ss[0]=='\0') break;
        insert_(ss);
    }
    while(scanf("%s",ss)!=EOF)
    {
        printf("%d\n",find_(ss));
    }
    return 0;
}