字典树

HDU1251 最基本的字典树

参考http://www.cnblogs.com/tanky_woo/archive/2010/09/24/1833717.html

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
using namespace std;
#define MAX 26
typedef struct Trie
{
    Trie *next[MAX];
    int v;
}Trie;
Trie root;

void creatTrie(char *str)
{
    int len=strlen(str);
    Trie *p= &root,*q;
    int i,j;
    for(i=0;i<len;i++)
    {
        int id=str[i]-'a';
        if(p->next[id]==NULL)
        {
            q=(Trie *)malloc(sizeof(root));
            q->v=1;
            for(j=0;j<26;j++)
            {
                q->next[j]=NULL;
            }
            p->next[id]=q;
            p=p->next[id];
        }
        else {
            p->next[id]->v++;
            p=p->next[id];
        }
    }

}

int findTrie(char *str)
{
    int len=strlen(str);
    Trie *p=&root;
    int i;
    for(i=0;i<len;i++)
    {
        int id=str[i]-'a';
        p=p->next[id];
        if(p==NULL)return 0;
    }
    return p->v;
}

int main()
{
    char str[15];
    int i;
    for(i=0;i<26;i++)
    {
        root.next[i]=NULL;
    }
    while(gets(str)&&str[0]!='\0')creatTrie(str);
    while(~scanf("%s",str))
    {
        int ans=findTrie(str);
        printf("%d\n",ans);
    }

    return 0;
}