字典树

今天刚开始学习字典树，我的第一篇随笔。

字典树，又称单词查找树，Trie树，是一种树形结构，典型应用是用于统计，排序和保存大量的字符串，所以经常被搜索引擎系统用于文本词频统计。它的优点是：利用字符串的公共前缀来节约存储空间，最大限度的减少无谓的字符串比较，查询效率比哈希表高。

 

以HDU1251为例    time：125MS，memory：44032K

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<stdlib.h>
using namespace std;
typedef struct node
{
    struct node *next[26];
    int cnt;//cnt记录个数它之后有几个分支； 
}trie;//定义字典树 
trie *root;//根节点 
void init()//根节点初始化 
{
    int i; 
    root=(trie*)malloc(sizeof(trie));   
    for(i=0;i<26;i++)   
        root->next[i]=0;   
    root->cnt=2; 
}
void insert(char *s)//插入新节点 
{
    trie *newnode,*p;//p为当前节点 
    p=root;
    int i=0;
    while(s[i])
    {
        int k=s[i++]-'a';//k表示下一个节点的序号 
        if(p->next[k]){p=p->next[k];p->cnt++;}//若下一个节点不为空，继续访问下一个节点，分支数+1 
        else//若下一个节点为空，新开节点，分支数=1 
        {
            newnode=(trie *)malloc(sizeof(trie));//开内存 
            for(int j=0;j<26;j++)newnode->next[j]=0;
            p->next[k]=newnode;
            p=newnode;
            p->cnt=1;
        }    
    }
}
int find(char *s)
{
    int i=0;
    trie *p;
    p=root;
    while(s[i])
    {
        int k=s[i++]-'a';
        if(p->next[k])p=p->next[k];//若下一个节点不为空，继续访问下一个节点 
        else return 0;    
    }
    return p->cnt;
}
int main()
{
    char s[11];
    int i;
    //freopen("int.txt","r",stdin);
    init();
    while(gets(s)&&strcmp(s,"")!=0)
    {
        insert(s);    
    }
    while(scanf("%s",s)!=EOF)
    {
        int ans=find(s);
        printf("%d\n",ans);    
    }
    return 0;
}