题目地址:http://acm.hdu.edu.cn/showproblem.php?pid=1251
可以说是一道模板题。
代码如下:
#include <iostream>
#include <vector>
#include <List>
#include <deque>
#include <queue>
#include <stack>
#include <map>
#include <set>
#include <algorithm>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <string>
#include <cmath>
using namespace std;
const int N=26;
typedef long long LL;
char xh[16];
struct Trie
{
Trie *next[N];//next是表示每层有多少种类的数,如果只是小写字母,则26即可,若改为大小写字母,则是52,若再加上数字,则是62了,这里根据题意来确定。
int v;//v可以表示一个字典树到此有多少相同前缀的数目,这里根据需要应当学会自由变化。
}*root;
void setroot()
{
root=(Trie *)malloc(sizeof(Trie));
for(int i=0;i<N;i++)
root->next[i] = NULL;
}
void createTrie(char *str)
{
int i,j,id;
int len=strlen(str);
Trie *p=root,*q;
for(i=0;i<len;i++)
{
id=str[i]-'a';//这个地方依据题意来定
if(p->next[id]==NULL)
{
q=(Trie *)malloc(sizeof(Trie));
q->v=1;//如果没有的话,初始为1
for(j=0;j<N;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 i,id;
int len=strlen(str);
Trie *p=root;
for(i=0;i<len;i++)
{
id=str[i]-'a';
p=p->next[id];
if(p==NULL) //若为空集,表示不存以此为前缀的串
return 0;
}
return p->v; //此串是字符集中某串的前缀
}
void dealTrie(Trie *T)
{//若果有多组数据-多棵树,那么要释放内纯
int i;
if(T==NULL)
return ;
for(i=0;i<N;i++)
if(T->next[i]!=NULL)
dealTrie(T->next[i]);
free(T);//释放
}
int main()
{
int i,j;
setroot();
while(gets(xh)&&xh[0]!='\0')
createTrie(xh);
memset(xh,0,sizeof(xh));
while(gets(xh)!=NULL)
{
printf("%d\n",findTrie(xh));
}
return 0;
}