hdu1251 字典树

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#define maxn 26//26个字母
struct trie
{
trie *next[maxn];//向下26个字母扩展
int v;//记录个数
};
trie *root;
void init()
{
root=(trie*)malloc(sizeof(trie));
for(int i=0;i<maxn;i++)
root->next[i]=NULL;
root->v=0;
}
void creattrie(char *str)
{
trie *p,*q;
p=root;
int i,j;
int len=strlen(str);
for(i=0;i<len;i++)
{
int id=str[i]-'a';
if(p->next[id]==NULL)
{
q = (trie *)malloc(sizeof(trie));
q->v=0;
for(j=0;j<maxn;j++)
q->next[j]=NULL;
p->next[id]=q;
}
p=p->next[id];
p->v++;//次数增加
}
}
int query(char *str)
{
int i,j;
int len=strlen(str);
trie *p=root;
for(i=0;i<len;i++)
{
int id=str[i]-'a';
if(p->next[id]==NULL)
return 0;
p=p->next[id];
}
return p->v;
}
int main()
{
char str[15];
int i,j;
init();
while(gets(str))
{
if(strcmp(str,"")==0)break;
creattrie(str);
}
while(gets(str))
{
if(strcmp(str,"")==0)break;
printf("%d\n",query(str));
}
}