字典树模板

Posted 不积跬步无以至千里,不积小流无以成江海

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了字典树模板相关的知识,希望对你有一定的参考价值。

字典树模板

#include<stdio.h>
#include<string.h>
#define kind 26//字母种类
struct Treenode//树的结点结构
{
    int count;//这个附加变量在本题中记录遍历到该结点形成的字符串出现的次数,在不同题中可记录不同的内容。
    Treenode *next[kind];//指向儿子结点
    Treenode()//每个结点的初始化
    {
        count=1;
        for(int i=0;i<kind;i++)
            next[i]=NULL;
    }
};
void insert(Treenode *&root,char *word)//向以root为根结点的树中插入串word
{
    Treenode *location=root;
    int i=0,branch=0;
    if(location==NULL) 
    {
        location=new Treenode();
        root=location;
    }
    while(word[i])
    {
        branch=word[i]-a;
        if(location->next[branch]) location->next[branch]->count++;//如果该字符存在,串数量加1
        else location->next[branch]=new Treenode();//如果不存在,建新结点
        i++;
        location=location->next[branch];
    }
}
int search(Treenode *&root,char *word)//查找,与插入类似
{
    Treenode *location=root;
    int i=0,branch=0,ans;
    if(location==NULL) return 0;
    while(word[i])
    {
        branch=word[i]-a;
        if(!location->next[branch]) return 0;
        i++;
        location=location->next[branch];
        ans=location->count;
    }
    return ans;
}
int main()
{
    char word[101][10];
    char ask[10];
    Treenode *root=NULL;
    int n,i;
    scanf("%d",&n);
       for(i=0;i<n;i++)
       {
           scanf("%s",word[i]);
        insert(root,word[i]);
       }
       while(scanf("%s",ask)!=EOF)
       {
          printf("%d\n",search(root,ask));
       }
    return 0;
}

 

以上是关于字典树模板的主要内容,如果未能解决你的问题,请参考以下文章

字典树数组模板

cf842D 01字典树|线段树 模板见hdu4825

Shortest Prefixes 字典树模板

字典树模板+HDU 1671 ( Phone List )(字典树)

字典树(模板)

字典树模板