这两天学了下字典树~
参考连接:
http://www.cnblogs.com/cherish_yimi/archive/2009/10/12/1581666.html
http://www.wutianqi.com/?p=1359
以及hdu的lcy老师的课件。
hdoj的1251求前缀字符串的个数,模版水过,参照dd牛模版
#include<stdio.h>//algo:trie tree
#include<string.h>
#include<stdlib.h>
#define MAX 26
typedef struct Trie{
Trie *next[MAX];
int flag;
}Trie;
Trie root;
void createTrie(char *str)
{
int len=strlen(str);
Trie *p=&root,*q;
for(int i=0;i<len;i++)
{
int id=str[i]-'a';
if(p->next[id]==NULL)
{
q=(Trie *)malloc(sizeof(Trie));
q->flag=1;//init
for(int j=0;j<MAX;j++)
q->next[j]=NULL;
p->next[id]=q;
p=p->next[id];
}
else
{
p->next[id]->flag++;
p=p->next[id];
}
}
}
int findTrie(char *str)
{
int len=strlen(str);
Trie *p=&root;
for(int i=0;i<len;i++)
{
int id=str[i]-'a';//根据需要减去0,或者‘a’或者'A'
p=p->next[id];
if(p==NULL)//若为空集,表示不存在以此为前缀的串
return 0;
}
return p->flag;// 此串是字符集中某串的前缀
}
int dealTrie(Trie* T)//释放空间
{
int i;
if(T==NULL) return 0;
for(i=0;i<MAX;i++)
{
if(T->next[i]!=NULL)
dealTrie(T->next[i]);
}
free(T);
return 0;
}
int main()
{
char str[15];
while(gets(str))
{
if(str[0]=='\0') break;
createTrie(str);
}
char aim[15];
while(~scanf("%s",aim))
{
int ans=findTrie(aim);
printf("%d\n",ans);
}
//dealTrie(&root);
return 0;
}