trie 字典树

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

永远热血沸腾

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

阅读量263

点赞数

CC 4.0 BY-SA版权

分类专栏：算法笔记

本文链接：https://blog.youkuaiyun.com/xinag578/article/details/48877103

算法笔记专栏收录该内容

21 篇文章

订阅专栏

本文详细介绍了如何使用C++实现Trie Tree（字典树），用于高效匹配和存储字符串。通过实例演示了从初始化到字符串插入、搜索等操作的全过程，并通过代码示例展示了具体实现。

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

const int sigm_size=26;
const int maxnode=400000+5;
int sz; //结点
int ch[maxnode][sigm_size],val[maxnode];

int idx(char c)
{
    return c-'a';
}

void init()
{
    sz=1;
    memset(ch[0],0,sizeof(ch[0]));//初始化根结点
}

void trie(char *s,int v)
{
    int c,i,j,u=0,n=strlen(s);
    for(i=0;i<n;i++)
    {
        c=idx(s[i]);
        if(!ch[u][c])//结点不存在
        {
            memset(ch[sz],0,sizeof(ch[sz]));
            val[sz]=0;
            ch[u][c]=sz++;//新建结点
        }
        u=ch[u][c];//往下走
    }
    val[u]=v;//字符串最后一个字符的附加信息
}

指针写法

#include<bits/stdc++.h>
using namespace std;
struct trie
{
    int cnt,num;
    int id;
    trie *son,*bro;
};

trie *rt=new trie;
int tot;
void settrie(char *s,int j)
{
    trie *p=rt,*t,*t1;
    int n=strlen(s),i,c;
    for(i=0; i<n; i++)
    {
        c=s[i]-'a';
        while(p->num!=-1&&p->num!=c)
        {
            t=p;
            p=p->bro;
            if(p->num>c)
            {
                p=t;
                break;
            }
        }
        //if(i==0&&c==3) printf("%d\n",++tot);
        if(p->num==-1)
        {
            p->cnt=1;
            p->num=c;
            t= new trie;
            t->id=-1;
            t->num=-1;
            t->cnt=0;
            p->bro=t;

            t= new trie;
            t->num=-1;
            t->cnt=0;
            t->id=-1;
            p->son=t;

            p->id=j;
            p=p->son;
        }
        else if(p->num==c)
        {
            if(p->id!=j) p->cnt++;
            p->id=j;
            p=p->son;
        }
        else
        {
            t= new trie;
            t->num=c;
            t->cnt=1;
            t->id=-1;
            t1= new trie;
            t1->id=-1;
            t1->num=-1;
            t1->cnt=0;
            t->son=t1;
            t->bro=p->bro;
            p->bro=t;
            p=t;
            p->id=j;
            p=p->son;
        }
    }
}


int main()
{
    int ans,c,n,i,j,len;
    char s[50];
    trie *p,*t;
    while(~scanf("%d",&n))
    {
        rt=new trie;
        rt->num=-1;
        rt->cnt=0;
        rt->id=-1;
        tot=0;
        for(i=0; i<n; i++)
        {
            scanf("%s",s);
            len=strlen(s);
            for(j=0; j<len; j++)
                settrie(s+j,i);
        }
        scanf("%d",&n);
        for(i=0; i<n; i++)
        {
            scanf("%s",s);
            len=strlen(s);
            p=rt;
            ans=0;
            for(j=0; j<len; j++)
            {
                c=s[j]-'a';
                while(p->num!=-1&&p->num!=c)
                {
                    t=p;
                    p=p->bro;
                    if(p->num>c)
                    {
                        p=t;
                        break;
                    }
                }
                if(p->num==c)
                {
                    ans=p->cnt;
                    p=p->son;
                }
                else break;
            }
            if(j!=len) ans=0;
            printf("%d\n",ans);
        }
    }
    return 0;
}