trie树

最新推荐文章于 2025-03-09 10:32:18 发布

原创最新推荐文章于 2025-03-09 10:32:18 发布 · 110 阅读

CC 4.0 BY-SA版权

本文详细介绍了一种使用C++实现的字典树（Trie）数据结构，该结构用于高效存储和检索字符串集合。文章首先引入了字典树的概念，随后通过具体的代码示例展示了如何构建字典树，包括插入字符串、查找字符串以及遍历树输出所有单词等功能。通过阅读本文，读者可以深入了解字典树的工作原理，并掌握其在实际编程中的应用。

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

#include<cstdlib>
#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
const int maxn=50000*50+5;

int m[maxn][52]={0},cnt=0,root=0,v[maxn]={0},n,mm;
char s[55];
int r(char ch)
{
    if(ch>='A'&&ch<='Z') return ch-'A';
    if(ch>='a'&ch<='z') return ch-'a'+26;
}
void in(int &now,int i,char *ch)//加入单词
{
    if(!now) now=++cnt;
    if(!ch[i])
    {
        v[now]++;
        return;
    }
    int d=r(ch[i]); 
    in(m[now][d],i+1,ch);
}
void init()
{
    scanf("%d%d",&n,&mm);
    char ch[55];
    for(int i=1;i<=n;i++)
    {
        scanf("%s",ch);
        in(root,0,ch);
    }
}
void work(int now,int i)//输出这棵树
{
    if(v[now]) 
    {
        s[i]=0;
        printf("%s\n",s);
    }
    for(int k=0;k<52;k++) if(m[now][k])
    {
        s[i]= k<26? k+'A':k-26+'a';
        work(m[now][k],i+1);
    }
}
int find(char *ch)//找单词
{
    int now=root;
    for(int i=0;ch[i];i++)
    {
        int d=r(ch[i]);
        if(!m[now][d]) return 0;
        now=m[now][d];
    }
    return v[now];
}
int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    init();
    work(root,0);
    printf("\n");
    char ch[55];
    int id;
    for(int i=1;i<=mm;i++)
    {
        scanf("%s",ch);
        printf("%d\n",find(ch));
    }
    return 0;
}