hihoCoder1014 Trie树

最新推荐文章于 2019-02-08 17:39:54 发布

原创最新推荐文章于 2019-02-08 17:39:54 发布 · 376 阅读

0 ·

CC 4.0 BY-SA版权

ACM-树专栏收录该内容

20 篇文章

订阅专栏

本文介绍了一种Trie树的基本实现方法，并通过一个C++模板提供了插入和搜索字符串的功能。文章详细展示了如何创建Trie树节点，如何插入新字符串到Trie树中，以及如何查询特定字符串在树中的出现次数。

基本的tire树加上模板

#include<iostream>
#include<math.h>
#include<cstdio>
#include<stdlib.h>
#include<cstring>
using namespace std;
#define MAX 26
typedef  struct tirenode
{

    int nCount; // 该节点前缀 出现的次数

    struct tirenode *next[MAX];//该节点的后续节点

} tirenode;

int allocp =0;
tirenode *creat()
{
    tirenode *tmp=(tirenode*)malloc(sizeof(tirenode));
    tmp->nCount=1;
    for(int i=0;i<MAX;i++)
        tmp->next[i]=NULL;
    return tmp;

}
void intsert(tirenode **proot,char *str)
{

    tirenode*tmp=*proot;
    int i=0,k;
    while(str[i])
    {
        k=str[i]-'a';
        if(tmp->next[k])
        {
            tmp->next[k]->nCount++;
        }
        else {
            tmp->next[k]=creat();
        }
        tmp=tmp->next[k];
        i++;
    }
}
int searcht(tirenode*root,char *str)
{

    if(root==NULL)
        return 0;
    tirenode*tmp=root;
    int i=0,k;
    while(str[i])
    {

        k=str[i]-'a';
        if(tmp->next[k])
        {
            tmp=tmp->next[k];
        }
        else return 0;
        i++;
            }
            return tmp->nCount;

}
int main()
{
    tirenode *root=creat();
    int t;
    char s[100100];
    scanf("%d%*c",&t);
    while(t--)
    {
        gets(s);
        intsert(&root,s);
    }
    scanf("%d%*c",&t);
    while(t--)
    {gets(s);

        printf("%d\n",searcht(root,s));
    }
    return 0;
}